darabos commited on
Commit
121923b
·
1 Parent(s): e8a8341

Automatically load *_ops.py.

Browse files
Files changed (1) hide show
  1. server/main.py +7 -7
server/main.py CHANGED
@@ -1,15 +1,15 @@
1
  import dataclasses
2
  import fastapi
3
  import pathlib
 
4
  from . import ops
5
  from . import workspace
6
- # Which boxes are available is currently configured here.
7
- # This is not the final solution.
8
- from . import lynxkite_ops
9
- from . import networkx_ops
10
- # from . import pytorch_model_ops
11
- from . import lynxscribe_ops
12
- from . import llm_ops
13
 
14
  app = fastapi.FastAPI()
15
 
 
1
  import dataclasses
2
  import fastapi
3
  import pathlib
4
+ import pkgutil
5
  from . import ops
6
  from . import workspace
7
+
8
+ here = pathlib.Path(__file__).parent
9
+ for _, name, _ in pkgutil.iter_modules([str(here)]):
10
+ if name.endswith('_ops') and not name.startswith('test_'):
11
+ print(f'Importing {name}')
12
+ __import__(f'server.{name}')
 
13
 
14
  app = fastapi.FastAPI()
15