fix

Author: robzor92

python/hopsworks/project.py

@@ -103,14 +103,17 @@ def created(self):
         """Timestamp when the project was created"""
         return self._created
 
-    def get_feature_store(self, name: str = None) -> feature_store.FeatureStore:
+    def get_feature_store(self, name: str = None, engine: str = None) -> feature_store.FeatureStore:
         """Connect to Project's Feature Store.
 
         Defaulting to the project name of default feature store. To get a
         shared feature store, the project name of the feature store is required.
 
         # Arguments
             name: Project name of the feature store.
+            engine: Which engine to use, `"spark"`, `"python"` or `"training"`.
+                Defaults to `"python"` when connected to https://c.app.hopsworks.ai/
+                See hsfs.Connection.connection documentation for more information.
         # Returns
             `hsfs.feature_store.FeatureStore`: The Feature Store API
         # Raises
@@ -120,8 +123,7 @@ def get_feature_store(self, name: str = None) -> feature_store.FeatureStore:
 
         _client = client.get_instance()
         if type(_client) == Client:  # If external client
-            engine = None
-            if _client._host == constants.HOSTS.APP_HOST:
+            if _client._host == constants.HOSTS.APP_HOST and engine is None:
                 engine = "python"
             return connection(
                 host=_client._host,
@@ -131,7 +133,7 @@ def get_feature_store(self, name: str = None) -> feature_store.FeatureStore:
                 engine=engine,
             ).get_feature_store(name)
         else:
-            return connection().get_feature_store(name)  # If internal client
+            return connection().get_feature_store(name, engine=engine)  # If internal client
 
     def get_model_registry(self):
         """Connect to Project's Model Registry API.