Upload folder using huggingface_hub

src/app.py

@@ -1,6 +1,11 @@
-from collections.abc import Generator
+import sys
 from pathlib import Path
-from typing import Annotated, Any
+from typing import Any, Generator, List
+
+if sys.version_info[:2] >= (3, 11):
+    from typing import Annotated
+else:
+    from typing_extensions import Annotated
 
 from fastapi import Depends, FastAPI, Request, status
 from fastapi.exceptions import RequestValidationError
@@ -28,7 +33,7 @@ def stream_mp4(mp4_path: Path) -> StreamingResponse:
     return StreamingResponse(content=iter_file(), media_type="video/mp4")
 
 
-ObservationType = list[Board]
+ObservationType = List[Board]
 
 
 class GridResponseType(BaseModel):

src/app_state.py

@@ -1,20 +1,25 @@
-from typing import Any
+from __future__ import annotations
+
+from typing import TYPE_CHECKING, Any
 
 import numpy as np
+from loguru import logger
 
-try:
-    from typing import Self
-except ImportError:
-    from typing_extensions import Self
+if TYPE_CHECKING:
+    import sys
 
-from loguru import logger
+    from litrl.env.connect_four import ConnectFour
+
+    if sys.version_info[:2] >= (3, 11):
+        from typing import Self
+    else:
+        from typing_extensions import Self
 
 from litrl import make_multiagent
 from litrl.algo.mcts.agent import MCTSAgent
 from litrl.algo.mcts.mcts_config import MCTSConfigBuilder
 from litrl.algo.mcts.rollout import VanillaRollout
 from litrl.common.agent import Agent, RandomMultiAgent
-from litrl.env.connect_four import ConnectFour
 from litrl.model.sac.multi_agent import OnnxSacDeterministicMultiAgent
 from src.typing import AgentType, CpuConfig, RolloutPolicy
 
@@ -34,7 +39,7 @@ class AppState:
         self.set_agent()  # TODO in properties setter.
         self.agent: Agent[Any, Any]
 
-    def __new__(cls: type["AppState"]) -> "AppState":
+    def __new__(cls: type[AppState]) -> AppState:  # noqa: PYI034
         if cls._instance is None:
             cls._instance = super().__new__(cls)
             cls._instance.setup()
@@ -76,6 +81,10 @@ class AppState:
     def get_action(self) -> int:
         return self.agent.get_action(env=self.env)
 
+    def inform_reset(self) -> None:
+        if isinstance(self.agent, MCTSAgent):
+            self.agent.inform_reset()
+
     def inform_action(self, action: int) -> None:
         """Update the agent's state as a result of external changes to the environment."""
         if isinstance(self.agent, MCTSAgent):

src/huggingface/get_environments.py

@@ -1,4 +1,9 @@
-from huggingface_hub import HfApi
+from __future__ import annotations
+
+from typing import TYPE_CHECKING
+
+if TYPE_CHECKING:
+    from huggingface_hub import HfApi
 
 from src.constants import ENV_RESULTS_FILE_DEPTH, MODEL_REPO, MODEL_REPO_TYPE
 

src/huggingface/get_files.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 from pathlib import Path
 
 import yaml

src/huggingface/huggingface_client.py

@@ -1,7 +1,7 @@
 from huggingface_hub import HfApi
 
-from .get_environments import get_environments
-from .get_files import get_mp4_paths
+from src.huggingface.get_environments import get_environments
+from src.huggingface.get_files import get_mp4_paths
 
 
 class HuggingFaceClient:

src/typing.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import enum
 
 from pydantic import BaseModel