Upload folder using huggingface_hub

server/UnitTestCaseGenerator_environment.py

@@ -21,9 +21,22 @@ import json
 from typing import Any, Optional
 from uuid import uuid4
 
-from openenv.core.env_server.interfaces import Environment
-from openenv.core.env_server.types import State
+# from openenv.core.env_server.interfaces import Environment
+# from openenv.core.env_server.types import State
 
+
+try:
+    from openenv.core.env_server.interfaces import Environment
+    from openenv.core.env_server.types import State
+except ImportError:
+    # Stub classes so the module can be imported without openenv installed
+    class Environment:
+        pass
+
+    class State:
+        def __init__(self, episode_id=None, step_count=0):
+            self.episode_id = episode_id
+            self.step_count = step_count
 try:
     from ..models import UnittestcasegeneratorAction, UnittestcasegeneratorObservation
 except ImportError:

server/app.py

@@ -77,10 +77,5 @@ def main(host: str = "0.0.0.0", port: int = 8000):
     uvicorn.run(app, host=host, port=port)
 
 
-if __name__ == "__main__":
-    import argparse
-
-    parser = argparse.ArgumentParser()
-    parser.add_argument("--port", type=int, default=8000)
-    args = parser.parse_args()
-    main(port=args.port)
+if __name__ == '__main__':
+    main()