Upload folder using huggingface_hub

client.py

@@ -1,7 +1,7 @@
 """Client for the Chess OpenEnv environment."""
 
 from dataclasses import dataclass
-from typing import Any, Dict, List, Optional
+from typing import Any, Dict, Optional
 
 import httpx
 
@@ -65,7 +65,7 @@ class ChessEnvClient:
         Returns:
             Initial observation of the board state
         """
-        payload = {}
+        payload: Dict[str, Any] = {}
         if seed is not None:
             payload["seed"] = seed
         if episode_id is not None:

examples/basic_usage.py

@@ -89,7 +89,7 @@ def demonstrate_illegal_move():
     illegal_action = ChessAction(move="e2e5")  # Can't move pawn 3 squares
     obs, reward, done = env.step(illegal_action)
 
-    print(f"Attempted illegal move: e2e5")
+    print("Attempted illegal move: e2e5")
     print(f"Reward: {reward}")  # Should be negative
     print(f"Error: {obs.metadata.get('error', 'None')}")
     print(f"Done: {done}")  # Game continues

examples/openenv_training.py

@@ -17,7 +17,7 @@ Usage:
 """
 
 import random
-from moonfish.rl import ChessEnvClient, ChessAction, make_env
+from moonfish.rl import ChessAction, make_env
 
 
 def random_policy(legal_moves: list[str]) -> str:

models.py

@@ -1,7 +1,7 @@
 """Data models for the Chess OpenEnv environment."""
 
 from dataclasses import dataclass, field
-from typing import Any, Dict, List, Optional, Union
+from typing import Any, Dict, List, Optional
 
 
 @dataclass

server/app.py

@@ -1,9 +1,7 @@
 """FastAPI server for the Chess OpenEnv environment."""
 
-import os
 from pathlib import Path
 from typing import Any, Dict, Optional
-from dataclasses import asdict
 
 import chess
 from fastapi import FastAPI, HTTPException
@@ -12,7 +10,7 @@ from fastapi.responses import FileResponse
 from pydantic import BaseModel
 
 from moonfish.lib import search_move
-from ..models import ChessAction, RewardConfig
+from ..models import ChessAction
 from .chess_environment import ChessEnvironment
 
 

server/chess_environment.py

@@ -2,7 +2,7 @@
 
 import random
 import uuid
-from typing import Any, Dict, List, Optional, Tuple
+from typing import Any, Dict, Optional, Tuple
 
 import chess