Book a Demo!
CoCalc Logo Icon
StoreFeaturesDocsShareSupportNewsAboutPoliciesSign UpSign In
iperov
GitHub Repository: iperov/deepfacelab
Path: blob/master/core/joblib/ThisThreadGenerator.py
628 views
1
class ThisThreadGenerator(object):
2
def __init__(self, generator_func, user_param=None):
3
super().__init__()
4
self.generator_func = generator_func
5
self.user_param = user_param
6
self.initialized = False
7
8
def __iter__(self):
9
return self
10
11
def __next__(self):
12
if not self.initialized:
13
self.initialized = True
14
self.generator_func = self.generator_func(self.user_param)
15
16
return next(self.generator_func)
17