Add minimal type hinting for hopsworks.login and project.get_feature_store + stop execution method

Author: vatj

python/hopsworks/__init__.py

@@ -13,6 +13,7 @@
 #   See the License for the specific language governing permissions and
 #   limitations under the License.
 #
+from __future__ import annotations
 
 import warnings
 import logging
@@ -25,6 +26,7 @@
 from hopsworks.client.exceptions import RestAPIError, ProjectException
 from hopsworks import version, constants, client
 from hopsworks.connection import Connection
+from hopsworks import project
 
 # Needs to run before import of hsml and hsfs
 warnings.filterwarnings(action="ignore", category=UserWarning, module=r".*psycopg2")
@@ -62,7 +64,7 @@ def login(
     project: str = None,
     api_key_value: str = None,
     api_key_file: str = None,
-):
+) -> project.Project:
     """Connect to [Serverless Hopsworks](https://app.hopsworks.ai) by calling the `hopsworks.login()` function with no arguments.
 
     ```python

python/hopsworks/core/execution_api.py

@@ -76,3 +76,21 @@ def _delete(self, job_name, id):
             id,
         ]
         _client._send_request("DELETE", path_params)
+
+    def _stop(self, job_name: str, id: int) -> None:
+        _client = client.get_instance()
+        path_params = [
+            "project",
+            self._project_id,
+            "jobs",
+            job_name,
+            "executions",
+            id,
+            "status",
+        ]
+        _client._send_request(
+            "PUT",
+            path_params=path_params,
+            data={"state": "stopped"},
+            headers={"Content-Type": "application/json"},
+        )

python/hopsworks/execution.py

@@ -212,6 +212,15 @@ def delete(self):
         """
         self._execution_api._delete(self._job.name, self.id)
 
+    def stop(self):
+        """Stop the execution
+        !!! danger "Potentially dangerous operation"
+            This operation stops the execution.
+        # Raises
+            `RestAPIError`.
+        """
+        self._execution_api._stop(self.job_name, self.id)
+
     def await_termination(self):
         """Wait until execution reaches terminal state
         # Raises

python/hopsworks/project.py

@@ -13,6 +13,7 @@
 #   See the License for the specific language governing permissions and
 #   limitations under the License.
 #
+from __future__ import annotations
 
 import humps
 import json
@@ -28,7 +29,7 @@
     environment_api,
     flink_cluster_api,
 )
-
+from hsfs import feature_store
 
 class Project:
     def __init__(
@@ -101,7 +102,7 @@ def created(self):
         """Timestamp when the project was created"""
         return self._created
 
-    def get_feature_store(self, name: str = None):
+    def get_feature_store(self, name: str = None) -> feature_store.FeatureStore:
         """Connect to Project's Feature Store.
 
         Defaulting to the project name of default feature store. To get a