Spaces:
Sleeping
Sleeping
Upload folder using huggingface_hub
Browse files- server/app.py +5 -3
server/app.py
CHANGED
|
@@ -38,11 +38,13 @@ except Exception as e: # pragma: no cover
|
|
| 38 |
try:
|
| 39 |
from ..models import UnittestcasegeneratorAction, UnittestcasegeneratorObservation
|
| 40 |
from .UnitTestCaseGenerator_environment import UnittestcasegeneratorEnvironment
|
| 41 |
-
except ModuleNotFoundError:
|
|
|
|
|
|
|
|
|
|
| 42 |
from models import UnittestcasegeneratorAction, UnittestcasegeneratorObservation
|
| 43 |
from server.UnitTestCaseGenerator_environment import UnittestcasegeneratorEnvironment
|
| 44 |
|
| 45 |
-
|
| 46 |
# Create the app with web interface and README integration
|
| 47 |
app = create_app(
|
| 48 |
UnittestcasegeneratorEnvironment,
|
|
@@ -81,4 +83,4 @@ if __name__ == "__main__":
|
|
| 81 |
parser = argparse.ArgumentParser()
|
| 82 |
parser.add_argument("--port", type=int, default=8000)
|
| 83 |
args = parser.parse_args()
|
| 84 |
-
main(port=args.port)
|
|
|
|
| 38 |
try:
|
| 39 |
from ..models import UnittestcasegeneratorAction, UnittestcasegeneratorObservation
|
| 40 |
from .UnitTestCaseGenerator_environment import UnittestcasegeneratorEnvironment
|
| 41 |
+
except (ImportError, ModuleNotFoundError):
|
| 42 |
+
import sys
|
| 43 |
+
import os
|
| 44 |
+
sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
|
| 45 |
from models import UnittestcasegeneratorAction, UnittestcasegeneratorObservation
|
| 46 |
from server.UnitTestCaseGenerator_environment import UnittestcasegeneratorEnvironment
|
| 47 |
|
|
|
|
| 48 |
# Create the app with web interface and README integration
|
| 49 |
app = create_app(
|
| 50 |
UnittestcasegeneratorEnvironment,
|
|
|
|
| 83 |
parser = argparse.ArgumentParser()
|
| 84 |
parser.add_argument("--port", type=int, default=8000)
|
| 85 |
args = parser.parse_args()
|
| 86 |
+
main(port=args.port)
|