Update evo_architecture.py

evo_architecture.py

@@ -1,10 +1,10 @@
-# evo_architecture.py
-
 import random
 import json
 import os
 import uuid
 import csv
+import torch
+import torch.nn as nn
 
 GENOME_LOG = "genome_log.csv"
 BEST_GENOME_FILE = "best_genome.json"
@@ -26,13 +26,9 @@ def default_config():
     }
 
 def mutate_genome(base_config, exploration_rate=0.5):
-    """Mutates the genome. Exploration rate controls how wild the mutation is (0 = minor, 1 = wild)."""
     config = base_config.copy()
     config["genome_id"] = str(uuid.uuid4())
-
-    # Select mutation type
     mutation_type = random.choice(["num_layers", "ffn_dim", "num_heads", "memory_enabled"])
-
     if mutation_type == "memory_enabled":
         config["memory_enabled"] = not config["memory_enabled"]
     else:
@@ -40,11 +36,9 @@ def mutate_genome(base_config, exploration_rate=0.5):
         change = int((max_val - min_val) * exploration_rate)
         delta = random.randint(-change, change)
         config[mutation_type] = max(min_val, min(max_val, config[mutation_type] + delta))
-
     return config
 
 def log_genome(config, score=None):
-    """Logs genome config + performance to CSV."""
     row = [
         config.get("genome_id", ""),
         config["num_layers"],
@@ -61,13 +55,21 @@ def log_genome(config, score=None):
         writer.writerow(row)
 
 def save_best_genome(config):
-    """Saves the best-performing genome to file."""
     with open(BEST_GENOME_FILE, "w", encoding="utf-8") as f:
         json.dump(config, f)
 
 def load_best_genome():
-    """Loads best-performing genome (if exists), else default."""
     if os.path.exists(BEST_GENOME_FILE):
         with open(BEST_GENOME_FILE, "r", encoding="utf-8") as f:
             return json.load(f)
     return default_config()
+
+# ✅ FIXED: This now exists and solves your import error
+def build_model_from_config(config):
+    from evo_model import EvoTransformerV22
+    return EvoTransformerV22(
+        num_layers=config["num_layers"],
+        ffn_dim=config["ffn_dim"],
+        num_heads=config["num_heads"],
+        memory_enabled=config["memory_enabled"]
+    )