PRANAV05092003 commited on
Commit
900e1f4
·
1 Parent(s): a92b800

Updated structure and fixed module import issue

Browse files
Files changed (3) hide show
  1. README.md +1 -0
  2. openenv_interface.py +6 -0
  3. server.py +4 -2
README.md CHANGED
@@ -3,6 +3,7 @@ title: ACRE - Autonomous Code Refactoring Environment
3
  colorFrom: blue
4
  colorTo: green
5
  sdk: docker
 
6
  app_port: 7860
7
  pinned: false
8
  license: mit
 
3
  colorFrom: blue
4
  colorTo: green
5
  sdk: docker
6
+ app_file: server.py
7
  app_port: 7860
8
  pinned: false
9
  license: mit
openenv_interface.py CHANGED
@@ -1,7 +1,13 @@
1
  from __future__ import annotations
2
 
 
 
3
  from typing import Any, Dict, Optional, Tuple
4
 
 
 
 
 
5
  try:
6
  from openenv.env import Env as OpenEnvBase
7
  except Exception: # pragma: no cover
 
1
  from __future__ import annotations
2
 
3
+ import os
4
+ import sys
5
  from typing import Any, Dict, Optional, Tuple
6
 
7
+ PROJECT_ROOT = os.path.dirname(os.path.abspath(__file__))
8
+ if PROJECT_ROOT not in sys.path:
9
+ sys.path.insert(0, PROJECT_ROOT)
10
+
11
  try:
12
  from openenv.env import Env as OpenEnvBase
13
  except Exception: # pragma: no cover
server.py CHANGED
@@ -12,10 +12,10 @@ Endpoints (all required by OpenEnv spec):
12
  from __future__ import annotations
13
 
14
  import difflib
15
- import sys
16
  import os
17
  import re
18
  import json
 
19
  from typing import Optional
20
 
21
  import uvicorn
@@ -25,7 +25,9 @@ from fastapi.middleware.cors import CORSMiddleware
25
  from fastapi.responses import HTMLResponse
26
  from openai import OpenAI
27
 
28
- sys.path.append(os.path.abspath("."))
 
 
29
 
30
  try:
31
  from stable_baselines3 import PPO
 
12
  from __future__ import annotations
13
 
14
  import difflib
 
15
  import os
16
  import re
17
  import json
18
+ import sys
19
  from typing import Optional
20
 
21
  import uvicorn
 
25
  from fastapi.responses import HTMLResponse
26
  from openai import OpenAI
27
 
28
+ PROJECT_ROOT = os.path.dirname(os.path.abspath(__file__))
29
+ if PROJECT_ROOT not in sys.path:
30
+ sys.path.insert(0, PROJECT_ROOT)
31
 
32
  try:
33
  from stable_baselines3 import PPO