Upload alpha_factory/deterministic/__init__.py

alpha_factory/deterministic/__init__.py

@@ -7,7 +7,9 @@ from .theme_sampler import (
     THEME_FIELDS, THEME_TO_ARCHETYPE, THEME_TO_TAG,
     get_theme_fields, get_theme_archetype, get_theme_tag,
 )
-from .fitness import compute_fitness, would_pass_brain, regime_tag_years
+from .fitness import compute_fitness, would_pass_brain
+from .proven_templates import generate_batch_from_proven_templates
+from .expression_mutator import generate_mutations
 
 # Backward-compat alias: list of known archetype names
 PROVEN_ARCHETYPES = list(THEME_TO_ARCHETYPE.values())
@@ -17,5 +19,6 @@ __all__ = [
     "pick_theme", "compute_gap_scores", "PROVEN_ARCHETYPES",
     "THEME_FIELDS", "THEME_TO_ARCHETYPE", "THEME_TO_TAG",
     "get_theme_fields", "get_theme_archetype", "get_theme_tag",
-    "compute_fitness", "would_pass_brain", "regime_tag_years",
+    "compute_fitness", "would_pass_brain",
+    "generate_batch_from_proven_templates", "generate_mutations",
 ]