Merge pull request #67 from iraikov/fix/epoch-index

Fix epoch index reference
iraikov · Dec 19, 2023 · 88ae403 · 88ae403
2 parents b565845 + e3baa9d
commit 88ae403
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/dmosopt/MOASMO.py b/dmosopt/MOASMO.py
@@ -43,7 +43,6 @@ def optimize(
     pct: percentage of resampled points in each iteration
     Xinit and Yinit: initial samplers for surrogate model construction
     """
-
     optimizer_kwargs = {}
     optimizer_kwargs.update(kwargs)
 

diff --git a/dmosopt/dmosopt.py b/dmosopt/dmosopt.py
@@ -336,7 +336,7 @@ def update_epoch(self, resample=False):
             else:
                 x_gen = item
                 for i in range(x_gen.shape[0]):
-                    self.append_request(EvalRequest(x_gen[i, :], None, epoch_index))
+                    self.append_request(EvalRequest(x_gen[i, :], None, self.epoch_index))
                 return_state = StrategyState.EnqueuedRequests
                 return_value = x_gen
 
@@ -376,7 +376,7 @@ def update_epoch(self, resample=False):
             else:
                 x_gen = item
                 for i in range(x_gen.shape[0]):
-                    self.append_request(EvalRequest(x_gen[i, :], None, epoch_index))
+                    self.append_request(EvalRequest(x_gen[i, :], None, self.epoch_index))
                 return_state = StrategyState.EnqueuedRequests
                 return_value = x_gen