Use model_name_suffix in _get_job_name() (#209)

spotify · Aug 27, 2019 · cbaa6b8 · cbaa6b8
1 parent 810d7de
commit cbaa6b8
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/spotify_tensorflow/luigi/tensorflow_task.py b/spotify_tensorflow/luigi/tensorflow_task.py
@@ -162,8 +162,11 @@ def _get_job_args(self):
         return args
 
     def _get_job_name(self):
-        job_name = "%s_%s_%s" % (getpass.getuser(), self.__class__.__name__,
-                                 str(uuid.uuid4()).replace("-", "_"))
+        job_name = "%s_%s_%s_%s" % (
+            getpass.getuser(),
+            self.__class__.__name__,
+            self.model_name_suffix,
+            str(uuid.uuid4()).replace("-", "_"))
         return job_name
 
     def _get_input_args(self):