From 8af568447f3bddee4e6dca97a06fd3c744915b7e Mon Sep 17 00:00:00 2001 From: Aleksey Veresov Date: Mon, 8 Jul 2024 17:11:45 +0200 Subject: [PATCH] Fix documentation generation --- python/hopsworks/project.py | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) diff --git a/python/hopsworks/project.py b/python/hopsworks/project.py index 294a69435..79ccff369 100644 --- a/python/hopsworks/project.py +++ b/python/hopsworks/project.py @@ -17,6 +17,7 @@ import json +import hsfs.feature_store import humps from hopsworks import client, constants, util from hopsworks.client.external import Client @@ -29,7 +30,6 @@ kafka_api, opensearch_api, ) -from hsfs import feature_store class Project: @@ -103,7 +103,9 @@ def created(self): """Timestamp when the project was created""" return self._created - def get_feature_store(self, name: str = None, engine: str = None) -> feature_store.FeatureStore: + def get_feature_store( + self, name: str = None, engine: str = None + ) -> hsfs.feature_store.FeatureStore: """Connect to Project's Feature Store. Defaulting to the project name of default feature store. To get a @@ -142,7 +144,9 @@ def get_feature_store(self, name: str = None, engine: str = None) -> feature_sto engine=engine, ).get_feature_store(name) else: - return connection(engine=engine).get_feature_store(name) # If internal client + return connection(engine=engine).get_feature_store( + name + ) # If internal client def get_model_registry(self): """Connect to Project's Model Registry API.