fix: update unused worker imports

shared/biosimulations_runutils/biosim_pipeline/biosim_api.py

@@ -6,8 +6,8 @@
 import requests
 from pydantic import BaseModel
 
-from datagen.biosimulations_runutils.biosim_pipeline.data_manager import DataManager
-from datagen.biosimulations_runutils.biosim_pipeline.datamodels import SourceOmex, Simulator, SimulationRun, BiosimulationsProject
+from shared.biosimulations_runutils.biosim_pipeline.data_manager import DataManager
+from shared.biosimulations_runutils.biosim_pipeline.datamodels import SimulationRun, SourceOmex, Simulator, BiosimulationsProject
 
 
 class _SimulationRunApiRequest(BaseModel):

worker/__init__.py

@@ -1,3 +1 @@
-from datagen.generator import *
-from datagen.data_model import *
-from datagen.utils import *
+

worker/data_generator.py

@@ -1,16 +1,13 @@
 import asyncio
 import time
 import os
-import uuid
 import warnings
 
 import requests
 from tempfile import mkdtemp
 from shutil import rmtree
-from typing import *
 from pathlib import Path
 
-import numpy as np
 from dotenv import load_dotenv
 
 from shared.biosimulations_runutils.biosim_pipeline.datamodels import Simulator

worker/main.py

@@ -4,7 +4,7 @@
 
 from dotenv import load_dotenv
 
-from shared.shared_worker import MongoDbConnector
+from shared.database import MongoDbConnector
 from shared.log_config import setup_logging
 from job import Supervisor