fix: remove infer need

Author: Justin Merrell

runpod/serverless/__init__.py

@@ -3,8 +3,8 @@
 from . import pod_worker
 
 
-def start():
+def start(config):
     '''
     Starts the serverless worker.
     '''
-    pod_worker.start_worker()
+    pod_worker.start_worker(config)

runpod/serverless/modules/job.py

@@ -6,7 +6,7 @@
 import requests
 
 from .logging import log
-from . import inference
+# from . import inference
 
 
 def get(worker_id):
@@ -51,17 +51,19 @@ def get(worker_id):
     return None
 
 
-def run(job):
+def run(job, run_handler):
     '''
     Run the job.
     Returns list of URLs and Job Time
     '''
     time_job_started = time.time()
     log(f"Started working on {job['id']} at {time_job_started} UTC")
 
-    model = inference.Model()
+    # model = inference.Model()
+
+    # job_output = model.run(job)
+    job_output = run_handler(job)
 
-    job_output = model.run(job)
     log(f"Job output: {job_output}")
 
     for index, output in enumerate(job_output):

runpod/serverless/pod_worker.py

@@ -10,7 +10,7 @@
 from .modules.logging import log
 
 
-def start_worker():
+def start_worker(config):
     '''
     Starts the worker.
     '''
@@ -30,7 +30,7 @@ def start_worker():
                     job.error(worker_life.worker_id, next_job['id'], "No input provided.")
                     continue
 
-                job_results = job.run(next_job)
+                job_results = job.run(next_job, config['handler'])
 
                 if 'error' in job_results:
                     job.error(worker_life.worker_id, next_job['id'], job_results['error'])

setup.cfg

@@ -1,6 +1,6 @@
 [metadata]
 name = runpod
-version = 0.4.6
+version = 0.5.0
 description = Official Python library for RunPod API & SDK.
 long_description = file: README.md
 long_description_content_type = text/markdown