fix: audit fixes - aegis_env/curriculum.py

aegis_env/curriculum.py

@@ -7,7 +7,10 @@ import json
 import random
 from typing import Dict, Any, Optional
 
-from scripts.adversarial_generator import AdversarialGenerator
+try:
+    from scripts.adversarial_generator import AdversarialGenerator as _AdversarialGenerator
+except ImportError:
+    _AdversarialGenerator = None
 
 MAX_TOKENS = 300  # Layer-3: token cap (~300 words)
 
@@ -85,9 +88,9 @@ class ScenarioLoader:
         self._step_counter += 1
         scenario = pool[idx]
         
-        if level >= 3:
-            return AdversarialGenerator.generate(scenario, seed=self.seed + self._step_counter)
-            
+        if level >= 3 and _AdversarialGenerator is not None:
+            return _AdversarialGenerator.generate(scenario, seed=self.seed + self._step_counter)
+
         return scenario
 
     def reset_counter(self) -> None: