Fix no local TF model accessible with LR enabled (#1668)

RobertLucian · web-flow · commit fd563114e891 · 2020-12-08T11:45:37.000-08:00
diff --git a/pkg/workloads/cortex/lib/model/cron.py b/pkg/workloads/cortex/lib/model/cron.py
@@ -1236,6 +1236,9 @@ def _load_local_models(self) -> None:
                             )
                         self._reset_when_tfs_unresponsive()
                         return None
+                self._old_ts_state[f"{model_name}-{model_version}"] = int(
+                    datetime.datetime.now(datetime.timezone.utc).timestamp()
+                )
 
     def _is_this_a_newer_model_id(self, model_id: str, timestamp: int) -> bool:
         return model_id in self._old_ts_state and self._old_ts_state[model_id] < timestamp

Original file line number	Diff line number	Diff line change
`@@ -1236,6 +1236,9 @@ def _load_local_models(self) -> None:`
`1236`	`1236`	`)`
`1237`	`1237`	`self._reset_when_tfs_unresponsive()`
`1238`	`1238`	`return None`
	`1239`	`+ self._old_ts_state[f"{model_name}-{model_version}"] = int(`
	`1240`	`+ datetime.datetime.now(datetime.timezone.utc).timestamp()`
	`1241`	`+ )`
`1239`	`1242`
`1240`	`1243`	`def _is_this_a_newer_model_id(self, model_id: str, timestamp: int) -> bool:`
`1241`	`1244`	`return model_id in self._old_ts_state and self._old_ts_state[model_id] < timestamp`