Skip to content

Commit

Permalink
Fix documentation generation
Browse files Browse the repository at this point in the history
  • Loading branch information
aversey committed Jul 14, 2024
1 parent 591fd0c commit 8af5684
Showing 1 changed file with 7 additions and 3 deletions.
10 changes: 7 additions & 3 deletions python/hopsworks/project.py
Original file line number Diff line number Diff line change
Expand Up @@ -17,6 +17,7 @@

import json

import hsfs.feature_store
import humps
from hopsworks import client, constants, util
from hopsworks.client.external import Client
Expand All @@ -29,7 +30,6 @@
kafka_api,
opensearch_api,
)
from hsfs import feature_store


class Project:
Expand Down Expand Up @@ -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
Expand Down Expand Up @@ -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.
Expand Down

0 comments on commit 8af5684

Please sign in to comment.