Merge pull request #36 from harrisonfloam/add-generator-imports

hiive · Sep 30, 2024 · 920dc6a · 920dc6a
2 parents 9c5d6ba + 916394b
commit 920dc6a
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/mlrose_hiive/__init__.py b/mlrose_hiive/__init__.py
@@ -27,7 +27,7 @@
 from .runners import GARunner, MIMICRunner, RHCRunner, SARunner, NNGSRunner, SKMLPRunner
 from .runners import (build_data_filename)
 from .generators import (MaxKColorGenerator, QueensGenerator, FlipFlopGenerator, TSPGenerator, KnapsackGenerator,
-                         ContinuousPeaksGenerator)
+                         ContinuousPeaksGenerator, FourPeaksGenerator)
 
 from .samples import SyntheticData
 from .samples import (plot_synthetic_dataset)

diff --git a/mlrose_hiive/generators/__init__.py b/mlrose_hiive/generators/__init__.py
@@ -9,3 +9,4 @@
 from .queens_generator import QueensGenerator
 from .tsp_generator import TSPGenerator
 from .continuous_peaks_generator import ContinuousPeaksGenerator
+from .four_peaks_generator import FourPeaksGenerator