Apply 2 suggestion(s) to 1 file(s)

arkindex_worker/worker/base.py

@@ -124,7 +124,8 @@ class BaseWorker(object):
             self.work_dir = os.path.join(xdg_data_home, "arkindex")
             os.makedirs(self.work_dir, exist_ok=True)
 
-        # Store task ID
+        # Store task ID. This is only available when running in production
+        # through a ponos agent
         self.task_id = os.environ.get("PONOS_TASK")
 
         self.worker_version_id = os.environ.get("WORKER_VERSION_ID")
@@ -364,7 +365,8 @@ class BaseWorker(object):
         :return: Path to the model on disk
         """
         if self.task_id:
-            # Running in PONOS
+            # When running in production with ponos, the agent
+            # downloads the model and set it in the current task work dir
             return Path(self.work_dir)
         else:
             model_path = self.config.get("model_dir", self.args.model_dir)