Reorder async refresh call

Stanley Opara · Stanley Opara · commit a264d643d65f · 2024-06-12T15:49:47.000-04:00
Signed-off-by: Stanley Opara &lt;a-sopara@expediagroup.com&gt;
diff --git a/sdk/python/feast/infra/registry/caching_registry.py b/sdk/python/feast/infra/registry/caching_registry.py
@@ -33,6 +33,7 @@ def __init__(self, project: str, cache_ttl_seconds: int, cache_mode: str):
         self.cache_mode = cache_mode
         if cache_mode == "thread":
             self._start_thread_async_refresh(cache_ttl_seconds)
+            atexit.register(self._exit_handler)
 
     @abstractmethod
     def _get_data_source(self, name: str, project: str) -> DataSource:
@@ -312,15 +313,14 @@ def _refresh_cached_registry_if_necessary(self):
                     self.refresh()
 
     def _start_thread_async_refresh(self, cache_ttl_seconds):
+        self.refresh()
         if cache_ttl_seconds <= 0:
             return
-        self.refresh()
         self.registry_refresh_thread = threading.Timer(
             cache_ttl_seconds, self._start_thread_async_refresh, [cache_ttl_seconds]
         )
         self.registry_refresh_thread.setDaemon(True)
         self.registry_refresh_thread.start()
-        atexit.register(self._exit_handler)
 
     def _exit_handler(self):
         self.registry_refresh_thread.cancel()
