Vidhikoul commited on
Commit
e87a121
·
verified ·
1 Parent(s): 88b7c69

Upload folder using huggingface_hub

Browse files
Files changed (1) hide show
  1. 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)