From 9f198f3b1556f411afda04b76359ce4ec0f47334 Mon Sep 17 00:00:00 2001 From: Steve Nyemba Date: Thu, 14 Apr 2022 12:03:59 -0500 Subject: [PATCH] bug fix: generator iherited by shuffle --- data/maker/__init__.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/data/maker/__init__.py b/data/maker/__init__.py index 49227a6..3f437d2 100644 --- a/data/maker/__init__.py +++ b/data/maker/__init__.py @@ -168,9 +168,12 @@ class Generator (Learner): self.network_args['candidates'] = int(_args['candidates']) if 'candidates' in _args else 1 filename = os.sep.join([self.network_args['logs'],'output',self.network_args['context'],'map.json']) self.log(**{'action':'init-map','input':{'filename':filename,'exists':os.path.exists(filename)}}) - file = open(filename) - self._map = json.loads(file.read()) - file.close() + if os.path.exists(filename): + file = open(filename) + self._map = json.loads(file.read()) + file.close() + else: + self._map = {} def run(self): self.initalize() if self._encoder is None :