mirror of
https://github.com/qurator-spk/eynollah.git
synced 2026-03-02 13:22:00 +01:00
try importing torch, then shapely, then tensorflow
This commit is contained in:
parent
44037bc05d
commit
f902756ce1
3 changed files with 32 additions and 22 deletions
|
|
@ -6,6 +6,8 @@ from typing import Union
|
|||
|
||||
import click
|
||||
|
||||
# NOTE: For debugging/predictable order of imports
|
||||
from .eynollah_imports import imported_libs
|
||||
from .model_zoo import EynollahModelZoo
|
||||
from .cli_models import models_cli
|
||||
|
||||
|
|
|
|||
8
src/eynollah/eynollah_imports.py
Normal file
8
src/eynollah/eynollah_imports.py
Normal file
|
|
@ -0,0 +1,8 @@
|
|||
"""
|
||||
Load libraries with possible race conditions once. This must be imported as the first module of eynollah.
|
||||
"""
|
||||
from torch import *
|
||||
import tensorflow.keras
|
||||
from shapely import *
|
||||
imported_libs = True
|
||||
__all__ = ['imported_libs']
|
||||
Loading…
Add table
Add a link
Reference in a new issue