synth-ai 0.2.4.dev6__py3-none-any.whl → 0.2.4.dev8__py3-none-any.whl

synth_ai/environments/examples/red/engine_helpers/reward_library/story_rewards.py

@@ -4,8 +4,9 @@ Story & Achievement Progression Reward Components
 Rewards for major milestones, story gates, and achievements.
 """
 
+from typing import Any, Dict, Set
+
 from synth_ai.environments.environment.rewards.core import RewardComponent
-from typing import Dict, Any, Set
 
 
 class GymBadgeEarnedReward(RewardComponent):

synth_ai/environments/examples/red/engine_helpers/screen_analysis.py

@@ -3,9 +3,10 @@ Screen analysis functions for Pokemon Red to provide rich textual descriptions
 of what's actually visible on the game screen.
 """
 
-import numpy as np
-from typing import Dict, Any, List, Tuple
 import hashlib
+from typing import Any, Dict, List, Tuple
+
+import numpy as np
 
 # Define some common Pokemon Red screen colors (RGB values)
 POKEMON_RED_COLORS = {

synth_ai/environments/examples/red/engine_helpers/state_extraction.py

@@ -1,5 +1,6 @@
+from typing import Any, Dict, List
+
 from .memory_map import *
-from typing import Dict, Any, List
 
 
 def get_byte(memory, addr: int) -> int:

synth_ai/environments/examples/red/environment.py

@@ -1,29 +1,32 @@
 from __future__ import annotations
-from typing import List, Optional, Any, Dict, Union
-from pydantic import BaseModel, Field
 
-# Import logging configuration to suppress JAX debug messages
+from typing import Any, Dict, List, Optional, Union
+
+from pydantic import BaseModel, Field
 
-from .engine import (
-    PokemonRedEngine,
-    PokemonRedPrivateState,
-    PokemonRedPublicState,
-    PokemonRedEngineSnapshot,
-)
-from .taskset import PokemonRedTaskInstance, INSTANCE as DEFAULT_TASK_INSTANCE
 from synth_ai.environments.environment.shared_engine import (
     GetObservationCallable,
     InternalObservation,
 )
-from synth_ai.environments.reproducibility.core import ReproducibleEnvironment
-from synth_ai.environments.stateful.core import StatefulEnvironment
 from synth_ai.environments.environment.tools import (
+    TOOL_REGISTRY,
     AbstractTool,
     EnvToolCall,
     ToolResult,
-    TOOL_REGISTRY,
     register_tool,
 )
+from synth_ai.environments.reproducibility.core import ReproducibleEnvironment
+from synth_ai.environments.stateful.core import StatefulEnvironment
+
+# Import logging configuration to suppress JAX debug messages
+from .engine import (
+    PokemonRedEngine,
+    PokemonRedEngineSnapshot,
+    PokemonRedPrivateState,
+    PokemonRedPublicState,
+)
+from .taskset import INSTANCE as DEFAULT_TASK_INSTANCE
+from .taskset import PokemonRedTaskInstance
 
 
 # Tool input schemas
@@ -73,9 +76,9 @@ class PokemonRedObservationCallable(GetObservationCallable):
     ) -> InternalObservation:
         """Convert Pokemon Red states to agent observation"""
         from .engine_helpers.state_extraction import (
-            get_badge_count,
-            format_position,
             format_hp_status,
+            format_position,
+            get_badge_count,
         )
 
         badge_count = get_badge_count(pub.badges)

synth_ai/environments/examples/red/taskset.py

@@ -1,12 +1,14 @@
 from __future__ import annotations
+
+import uuid
 from dataclasses import dataclass
 from pathlib import Path
-import uuid
+
 from synth_ai.environments.tasks.core import (
-    Task,
-    TaskInstance,
     Impetus,
     Intent,
+    Task,
+    TaskInstance,
     TaskInstanceMetadata,
 )
 

synth_ai/environments/examples/sokoban/engine.py

@@ -1,25 +1,28 @@
 from __future__ import annotations
-from typing import Optional, Dict, Any, Tuple
 
+import logging
+from dataclasses import dataclass
+from typing import Any, Dict, Optional, Tuple
+
+import numpy as np
+
+from synth_ai.environments.environment.rewards.core import RewardComponent, RewardStack
 from synth_ai.environments.environment.shared_engine import (
     GetObservationCallable,
     InternalObservation,
 )
-from synth_ai.environments.stateful.engine import StatefulEngine, StatefulEngineSnapshot
-from synth_ai.environments.tasks.core import TaskInstance
-import numpy as np
-from dataclasses import dataclass
-from synth_ai.environments.examples.sokoban.taskset import (
-    SokobanTaskInstance,
-)  # Assuming this is where SokobanTaskInstance is defined
-from synth_ai.environments.reproducibility.core import IReproducibleEngine  # Added import
-import logging
-from synth_ai.environments.environment.rewards.core import RewardStack, RewardComponent
 from synth_ai.environments.examples.sokoban.engine_helpers.vendored.envs.sokoban_env import (
     ACTION_LOOKUP,
+)
+from synth_ai.environments.examples.sokoban.engine_helpers.vendored.envs.sokoban_env import (
     SokobanEnv as GymSokobanEnv,
 )
-import numpy as np
+from synth_ai.environments.examples.sokoban.taskset import (
+    SokobanTaskInstance,
+)  # Assuming this is where SokobanTaskInstance is defined
+from synth_ai.environments.reproducibility.core import IReproducibleEngine  # Added import
+from synth_ai.environments.stateful.engine import StatefulEngine, StatefulEngineSnapshot
+from synth_ai.environments.tasks.core import TaskInstance
 
 # No monkey-patch needed - we fixed the vendored code directly
 
@@ -623,8 +626,8 @@ if __name__ == "__main__":
             "num_env_steps": 0,
         }
     }
-    import random
     import asyncio
+    import random
 
     async def sanity():
         task_instance = TaskInstance()

synth_ai/environments/examples/sokoban/engine_helpers/room_utils.py

@@ -1,9 +1,10 @@
-import numpy as np
-import marshal
 import copy
+import marshal
 from collections import deque
 from typing import Tuple
 
+import numpy as np
+
 # --- ADDED CONSTANT ---
 MAX_GENERATION_RETRIES = 100
 # --- END ADDED CONSTANT ---

synth_ai/environments/examples/sokoban/engine_helpers/vendored/__init__.py

@@ -1,6 +1,7 @@
+import json
 import logging
+
 import pkg_resources
-import json
 from gym.envs.registration import register
 
 logger = logging.getLogger(__name__)

synth_ai/environments/examples/sokoban/engine_helpers/vendored/envs/__init__.py

@@ -1,3 +1,3 @@
-from .sokoban_env import SokobanEnv, ACTION_LOOKUP, CHANGE_COORDINATES
 from . import room_utils
+from .sokoban_env import ACTION_LOOKUP, CHANGE_COORDINATES, SokobanEnv
 from .sokoban_env_variations import *

synth_ai/environments/examples/sokoban/engine_helpers/vendored/envs/boxoban_env.py

@@ -1,13 +1,15 @@
-from .sokoban_env import SokobanEnv
-from .render_utils import room_to_rgb
 import os
+import random
+import zipfile
 from os import listdir
 from os.path import isfile, join
+
+import numpy as np
 import requests
-import zipfile
 from tqdm import tqdm
-import random
-import numpy as np
+
+from .render_utils import room_to_rgb
+from .sokoban_env import SokobanEnv
 
 
 class BoxobanEnv(SokobanEnv):

synth_ai/environments/examples/sokoban/engine_helpers/vendored/envs/render_utils.py

@@ -1,6 +1,6 @@
+import imageio
 import numpy as np
 import pkg_resources
-import imageio
 
 
 def room_to_rgb(room, room_structure=None):

synth_ai/environments/examples/sokoban/engine_helpers/vendored/envs/room_utils.py

@@ -1,6 +1,7 @@
+import marshal
 import random
+
 import numpy as np
-import marshal
 
 
 def generate_room(

synth_ai/environments/examples/sokoban/engine_helpers/vendored/envs/sokoban_env.py

@@ -1,10 +1,11 @@
 import gym
-from gym.utils import seeding
-from gym.spaces.discrete import Discrete
+import numpy as np
 from gym.spaces import Box
-from .room_utils import generate_room
+from gym.spaces.discrete import Discrete
+from gym.utils import seeding
+
 from .render_utils import room_to_rgb, room_to_tiny_world_rgb
-import numpy as np
+from .room_utils import generate_room
 
 
 class SokobanEnv(gym.Env):

synth_ai/environments/examples/sokoban/engine_helpers/vendored/envs/sokoban_env_fixed_targets.py

@@ -1,7 +1,8 @@
-from .sokoban_env import SokobanEnv
-from .render_utils import room_to_rgb_FT, room_to_tiny_world_rgb_FT
 from gym.spaces import Box
 
+from .render_utils import room_to_rgb_FT, room_to_tiny_world_rgb_FT
+from .sokoban_env import SokobanEnv
+
 
 class FixedTargetsSokobanEnv(SokobanEnv):
     def __init__(self, dim_room=(10, 10), max_steps=120, num_boxes=3, num_gen_steps=None):

synth_ai/environments/examples/sokoban/engine_helpers/vendored/envs/sokoban_env_pull.py

@@ -1,7 +1,8 @@
-from .sokoban_env import SokobanEnv, CHANGE_COORDINATES
 from gym.spaces import Box
 from gym.spaces.discrete import Discrete
 
+from .sokoban_env import CHANGE_COORDINATES, SokobanEnv
+
 
 class PushAndPullSokobanEnv(SokobanEnv):
     def __init__(self, dim_room=(10, 10), max_steps=120, num_boxes=3, num_gen_steps=None):

synth_ai/environments/examples/sokoban/engine_helpers/vendored/envs/sokoban_env_two_player.py

@@ -1,13 +1,14 @@
-from .sokoban_env import SokobanEnv, CHANGE_COORDINATES
+import numpy as np
 from gym.spaces import Box
 from gym.spaces.discrete import Discrete
+
 from .render_utils import (
-    room_to_rgb,
-    room_to_tiny_world_rgb,
     color_player_two,
     color_tiny_player_two,
+    room_to_rgb,
+    room_to_tiny_world_rgb,
 )
-import numpy as np
+from .sokoban_env import CHANGE_COORDINATES, SokobanEnv
 
 
 class TwoPlayerSokobanEnv(SokobanEnv):

synth_ai/environments/examples/sokoban/engine_helpers/vendored/envs/sokoban_env_variations.py

@@ -1,8 +1,8 @@
+from .boxoban_env import BoxobanEnv
 from .sokoban_env import SokobanEnv
 from .sokoban_env_fixed_targets import FixedTargetsSokobanEnv
 from .sokoban_env_pull import PushAndPullSokobanEnv
 from .sokoban_env_two_player import TwoPlayerSokobanEnv
-from .boxoban_env import BoxobanEnv
 
 
 class SokobanEnv1(SokobanEnv):

synth_ai/environments/examples/sokoban/environment.py

@@ -1,29 +1,30 @@
-from typing import List, Optional, Any, Dict, Union
-from pydantic import BaseModel
 import dataclasses
+from typing import Any, Dict, List, Optional, Union
+
+from pydantic import BaseModel
 
-from synth_ai.environments.examples.sokoban.engine import (
-    SokobanEngine,
-    SynthSokobanObservationCallable,
-    SokobanPrivateState,
-    SokobanPublicState,
-    SynthSokobanCheckpointObservationCallable,
-    SokobanEngineSnapshot,
-)
 from synth_ai.environments.environment.shared_engine import (
     GetObservationCallable,
     InternalObservation,
 )
-from synth_ai.environments.reproducibility.core import ReproducibleEnvironment
-from synth_ai.environments.stateful.core import StatefulEnvironment
-from synth_ai.environments.tasks.core import TaskInstance
 from synth_ai.environments.environment.tools import (
+    TOOL_REGISTRY,
     AbstractTool,
     EnvToolCall,
     ToolResult,
-    TOOL_REGISTRY,
     register_tool,
 )
+from synth_ai.environments.examples.sokoban.engine import (
+    SokobanEngine,
+    SokobanEngineSnapshot,
+    SokobanPrivateState,
+    SokobanPublicState,
+    SynthSokobanCheckpointObservationCallable,
+    SynthSokobanObservationCallable,
+)
+from synth_ai.environments.reproducibility.core import ReproducibleEnvironment
+from synth_ai.environments.stateful.core import StatefulEnvironment
+from synth_ai.environments.tasks.core import TaskInstance
 
 
 # --- Tool Definition ---

synth_ai/environments/examples/sokoban/generate_verified_puzzles.py

@@ -8,10 +8,12 @@ and saves them as JSON. Each puzzle is verified to be solvable using BFS.
 
 import json
 import logging
-import numpy as np
-from typing import Dict, List, Tuple, Optional, Any, Set
+from dataclasses import asdict, dataclass
 from pathlib import Path
-from dataclasses import dataclass, asdict
+from typing import Any, Dict, List, Optional, Set, Tuple
+
+import numpy as np
+
 from synth_ai.environments.examples.sokoban.engine_helpers.room_utils import (
     generate_room,
     get_shortest_action_path,

synth_ai/environments/examples/sokoban/puzzle_loader.py

@@ -5,9 +5,10 @@ Puzzle loader for pre-generated verified Sokoban puzzles.
 import json
 import logging
 import random
-from pathlib import Path
-from typing import Dict, List, Optional, Tuple, Any
 from dataclasses import dataclass
+from pathlib import Path
+from typing import Any, Dict, List, Optional, Tuple
+
 import numpy as np
 
 logger = logging.getLogger(__name__)

synth_ai/environments/examples/sokoban/taskset.py

@@ -1,20 +1,23 @@
+import logging
+import os
+from dataclasses import asdict, dataclass, fields
+from typing import List, Tuple
+from uuid import UUID, uuid4
+
+from synth_ai.environments.examples.sokoban.puzzle_loader import (
+    SokobanPuzzle,
+    get_puzzle_loader,
+)
 from synth_ai.environments.tasks.core import (
+    Impetus,
+    Intent,
+    SplitInfo,
     Task,
     TaskInstance,
     TaskInstanceMetadata,
     TaskInstanceMetadataFilter,
     TaskInstanceSet,
 )
-from uuid import uuid4, UUID
-from synth_ai.environments.tasks.core import SplitInfo, Impetus, Intent
-from synth_ai.environments.examples.sokoban.puzzle_loader import (
-    get_puzzle_loader,
-    SokobanPuzzle,
-)
-from dataclasses import dataclass, asdict, fields
-from typing import Tuple, List
-import os
-import logging
 
 logger = logging.getLogger(__name__)
 

synth_ai/environments/examples/tictactoe/engine.py

@@ -1,19 +1,19 @@
 from __future__ import annotations
 
-import numpy as np
 from dataclasses import dataclass
-from typing import Dict, Any, Optional, Tuple
+from typing import Any, Dict, Optional, Tuple
 
-from synth_ai.environments.stateful.engine import StatefulEngine, StatefulEngineSnapshot
-from synth_ai.environments.reproducibility.core import IReproducibleEngine
-from synth_ai.environments.environment.rewards.core import RewardStack, RewardComponent
+import numpy as np
+
+from synth_ai.environments.environment.rewards.core import RewardComponent, RewardStack
 from synth_ai.environments.environment.shared_engine import (
     GetObservationCallable,
     InternalObservation,
 )
+from synth_ai.environments.reproducibility.core import IReproducibleEngine
+from synth_ai.environments.stateful.engine import StatefulEngine, StatefulEngineSnapshot
 from synth_ai.environments.tasks.core import TaskInstance
 
-
 # Action mapping: coordinate strings to board indices
 COORD_TO_IDX = {
     "A1": 0,

synth_ai/environments/examples/tictactoe/environment.py

@@ -1,10 +1,9 @@
 from __future__ import annotations
 
-from typing import Dict, Optional, Any, List, Union
+from typing import Any, Dict, List, Optional, Union
+
 from pydantic import BaseModel
 
-from synth_ai.environments.stateful.core import StatefulEnvironment
-from synth_ai.environments.reproducibility.core import ReproducibleEnvironment
 from synth_ai.environments.environment.shared_engine import (
     GetObservationCallable,
     InternalObservation,
@@ -14,15 +13,17 @@ from synth_ai.environments.environment.tools import (
     EnvToolCall,
     ToolResult,
 )
+from synth_ai.environments.reproducibility.core import ReproducibleEnvironment
+from synth_ai.environments.stateful.core import StatefulEnvironment
 from synth_ai.environments.tasks.core import TaskInstance
 
 from .engine import (
+    SynthTicTacToeCheckpointObservationCallable,
+    SynthTicTacToeObservationCallable,
     TicTacToeEngine,
-    TicTacToePublicState,
-    TicTacToePrivateState,
     TicTacToeEngineSnapshot,
-    SynthTicTacToeObservationCallable,
-    SynthTicTacToeCheckpointObservationCallable,
+    TicTacToePrivateState,
+    TicTacToePublicState,
 )
 
 

synth_ai/environments/examples/tictactoe/taskset.py

@@ -3,19 +3,20 @@ from __future__ import annotations
 import random
 from dataclasses import dataclass
 from typing import List
+from uuid import uuid4
+
 import numpy as np
 
-from uuid import uuid4
 from synth_ai.environments.tasks.core import (
-    TaskInstance,
-    TaskInstanceMetadata,
-    TaskInstanceSet,
     Impetus,
     Intent,
     SplitInfo,
+    TaskInstance,
+    TaskInstanceMetadata,
+    TaskInstanceSet,
 )
 
-from .engine import COORD_TO_IDX, WIN_PATTERNS, PLAYER_MARKS
+from .engine import COORD_TO_IDX, PLAYER_MARKS, WIN_PATTERNS
 
 
 @dataclass

synth_ai/environments/examples/verilog/engine.py

@@ -1,13 +1,14 @@
 from __future__ import annotations
+
 import shutil
 import subprocess
-from pathlib import Path
-from typing import Dict, Any, Tuple, Optional
 from dataclasses import dataclass
+from pathlib import Path
+from typing import Any, Dict, Optional, Tuple
 
+from synth_ai.environments.environment.rewards.core import RewardComponent, RewardStack
 from synth_ai.environments.stateful.engine import StatefulEngine, StatefulEngineSnapshot
 from synth_ai.environments.tasks.core import TaskInstance
-from synth_ai.environments.environment.rewards.core import RewardStack, RewardComponent
 
 
 @dataclass

synth_ai/environments/examples/verilog/environment.py

@@ -1,25 +1,26 @@
-from typing import List, Optional, Any, Dict, Union
+from typing import Any, Dict, List, Optional, Union
+
 from pydantic import BaseModel
 
-from synth_ai.environments.examples.verilog.engine import (
-    VerilogEngine,
-    VerilogPrivateState,
-    VerilogPublicState,
-    VerilogEngineSnapshot,
-)
 from synth_ai.environments.environment.shared_engine import (
     GetObservationCallable,
     InternalObservation,
 )
-from synth_ai.environments.stateful.core import StatefulEnvironment
-from synth_ai.environments.tasks.core import TaskInstance
 from synth_ai.environments.environment.tools import (
+    TOOL_REGISTRY,
     AbstractTool,
     EnvToolCall,
     ToolResult,
-    TOOL_REGISTRY,
     register_tool,
 )
+from synth_ai.environments.examples.verilog.engine import (
+    VerilogEngine,
+    VerilogEngineSnapshot,
+    VerilogPrivateState,
+    VerilogPublicState,
+)
+from synth_ai.environments.stateful.core import StatefulEnvironment
+from synth_ai.environments.tasks.core import TaskInstance
 
 
 # Tool Input Schemas

synth_ai/environments/examples/verilog/taskset.py

@@ -1,21 +1,23 @@
+import atexit
+import os
+import shutil
+import tempfile
+from dataclasses import asdict, dataclass, fields
+from pathlib import Path
+from typing import Optional
+from uuid import UUID, uuid4
+
+from datasets import load_dataset
+
 from synth_ai.environments.tasks.core import (
+    Impetus,
+    Intent,
+    SplitInfo,
     Task,
     TaskInstance,
     TaskInstanceMetadata,
     TaskInstanceSet,
-    SplitInfo,
-    Impetus,
-    Intent,
 )
-from uuid import uuid4, UUID
-from dataclasses import dataclass, asdict, fields
-from typing import Optional
-from pathlib import Path
-import tempfile
-import os
-import shutil
-import atexit
-from datasets import load_dataset
 
 # Global list to track temp directories for cleanup
 _temp_dirs = []

synth_ai/environments/examples/wordle/__init__.py

@@ -1,12 +1,12 @@
 from .engine import (
+    SynthWordleCheckpointObservationCallable,
+    SynthWordleObservationCallable,
     WordleEngine,
-    WordlePublicState,
-    WordlePrivateState,
     WordleEngineSnapshot,
-    SynthWordleObservationCallable,
-    SynthWordleCheckpointObservationCallable,
+    WordlePrivateState,
+    WordlePublicState,
 )
-from .environment import WordleEnvironment, WordleInteractTool, WordleActionInput
+from .environment import WordleActionInput, WordleEnvironment, WordleInteractTool
 from .taskset import WordleTaskInstance, WordleTaskInstanceMetadata, create_wordle_taskset, taskset
 
 __all__ = [