[Feature store] Support deployment of an empty graph ingestion service (

mlrun#932)
Penumbra69 · May 15, 2021 · a8a438a · a8a438a
1 parent 5901bf0
commit a8a438a
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 7 deletions.
diff --git a/mlrun/datastore/targets.py b/mlrun/datastore/targets.py
@@ -274,7 +274,7 @@ def _get_store(self):
         return store
 
     def _get_column_list(self, features, timestamp_key, key_columns):
-        column_list = None
+        column_list = []
         if self.columns:
             return self.columns
         elif features:

diff --git a/mlrun/runtimes/serving.py b/mlrun/runtimes/serving.py
@@ -452,7 +452,7 @@ def _get_runtime_env(self):
             "function_uri": self._function_uri(),
             "version": "v2",
             "parameters": self.spec.parameters,
-            "graph": self.spec.graph.to_dict(),
+            "graph": self.spec.graph.to_dict() if self.spec.graph else {},
             "load_mode": self.spec.load_mode,
             "functions": function_name_uri_map,
             "graph_initializer": self.spec.graph_initializer,

diff --git a/mlrun/serving/states.py b/mlrun/serving/states.py
@@ -774,11 +774,6 @@ def has_loop(state, previous):
         if self.on_error and self.on_error in start_states:
             start_states.remove(self.on_error)
 
-        if (
-            not start_states
-        ):  # for safety, not sure if its possible to get here (since its a loop)
-            raise GraphError("there are no starting states (ones without .after)")
-
         if (
             len(responders) > 1
         ):  # should not have multiple steps which respond to request