Skip to content

Commit

Permalink
remove: skeleton methods
Browse files Browse the repository at this point in the history
  • Loading branch information
morgandu committed Nov 24, 2021
1 parent 254f7bd commit 4b35fcd
Show file tree
Hide file tree
Showing 3 changed files with 2 additions and 184 deletions.
99 changes: 1 addition & 98 deletions google/cloud/aiplatform/_featurestores/entity_type.py
Expand Up @@ -15,9 +15,7 @@
# limitations under the License.
#

from typing import Dict, List, Optional, Sequence, Tuple, Union

import datetime
from typing import Dict, List, Optional, Sequence, Tuple

from google.auth import credentials as auth_credentials
from google.protobuf import field_mask_pb2
Expand All @@ -28,13 +26,6 @@
from google.cloud.aiplatform import utils
from google.cloud.aiplatform.utils import featurestore_utils

try:
import pandas as pd
except ImportError:
raise ImportError(
"Pandas is not installed. Please install pandas to use Vertex Feature Store"
)

_LOGGER = base.Logger(__name__)
_ALL_FEATURE_IDS = "*"

Expand Down Expand Up @@ -412,91 +403,3 @@ def delete_features(
feature.delete(sync=sync)
if not sync:
feature.wait()

@classmethod
def create(
cls,
entity_type_id: str,
featurestore: str,
description: Optional[str] = None,
labels: Optional[Dict[str, str]] = None,
project: Optional[str] = None,
location: Optional[str] = None,
credentials: Optional[auth_credentials.Credentials] = None,
request_metadata: Optional[Sequence[Tuple[str, str]]] = (),
sync: Optional[bool] = True,
) -> "EntityType":
""""""
raise NotImplementedError

def create_feature(
self,
feature_id: str,
value_type: str,
description: Optional[str] = None,
labels: Optional[Dict[str, str]] = None,
request_metadata: Optional[Sequence[Tuple[str, str]]] = (),
sync: Optional[bool] = True,
) -> "_featurestores.Feature":
""""""
raise NotImplementedError

def batch_create_features(
self,
feature_configs: List[Dict[str, Union[bool, int, Dict[str, str], str]]],
request_metadata: Optional[Sequence[Tuple[str, str]]] = (),
sync: Optional[bool] = True,
) -> "EntityType":
""""""
raise NotImplementedError

def read(
self, entity_ids: List[str], feature_ids: Optional[List[str]] = None,
) -> pd.DataFrame:
""""""
raise NotImplementedError

def ingest_from_bq(
self,
bq_source_uri: str,
feature_ids: List[str],
source_feature_mapping: Optional[Dict[str, str]] = None,
entity_id_field: Optional[str] = "entity_id",
feature_time_field: Optional[str] = None,
feature_time: Optional[datetime.datetime] = None,
disable_online_serving: Optional[bool] = False,
worker_count: Optional[int] = 1,
sync: Optional[bool] = True,
) -> "EntityType":
""""""
raise NotImplementedError

def ingest_from_gcs(
self,
gcs_source_uris: Union[str, List[str]],
feature_ids: List[str],
source_feature_mapping: Optional[Dict[str, str]] = None,
entity_id_field: Optional[str] = "entity_id",
feature_time_field: Optional[str] = None,
feature_time: Optional[datetime.datetime] = None,
disable_online_serving: Optional[bool] = False,
worker_count: Optional[int] = 1,
sync: Optional[bool] = True,
) -> "EntityType":
""""""
raise NotImplementedError

def ingest_from_df(
self,
df_source: pd.DataFrame,
feature_ids: List[str],
source_feature_mapping: Optional[Dict[str, str]] = None,
entity_id_field: Optional[str] = "entity_id",
feature_time_field: Optional[str] = None,
feature_time: Optional[datetime.datetime] = None,
disable_online_serving: Optional[bool] = False,
worker_count: Optional[int] = 1,
sync: Optional[bool] = True,
) -> "EntityType":
""""""
raise NotImplementedError
18 changes: 0 additions & 18 deletions google/cloud/aiplatform/_featurestores/feature.py
Expand Up @@ -327,21 +327,3 @@ def list(
credentials=credentials,
parent=entity_type_name,
)

@classmethod
def create(
cls,
feature_id: str,
value_type: str,
entity_type_name: str,
featurestore_id: Optional[str] = None,
description: Optional[str] = None,
labels: Optional[Dict[str, str]] = None,
project: Optional[str] = None,
location: Optional[str] = None,
credentials: Optional[auth_credentials.Credentials] = None,
request_metadata: Optional[Sequence[Tuple[str, str]]] = (),
sync: Optional[bool] = True,
) -> "Feature":
""""""
raise NotImplementedError
69 changes: 1 addition & 68 deletions google/cloud/aiplatform/_featurestores/featurestore.py
Expand Up @@ -15,7 +15,7 @@
# limitations under the License.
#

from typing import Dict, List, Optional, Sequence, Tuple, Union
from typing import Dict, List, Optional, Sequence, Tuple

from google.auth import credentials as auth_credentials
from google.protobuf import field_mask_pb2
Expand All @@ -27,13 +27,6 @@
from google.cloud.aiplatform import utils
from google.cloud.aiplatform.utils import featurestore_utils

try:
import pandas as pd
except ImportError:
raise ImportError(
"Pandas is not installed. Please install pandas to use Vertex Feature Store"
)

_LOGGER = base.Logger(__name__)


Expand Down Expand Up @@ -485,63 +478,3 @@ def search_features(
)
for gapic_resource in feature_list
]

@classmethod
def create(
cls,
featurestore_id: str,
labels: Optional[Dict[str, str]] = None,
project: Optional[str] = None,
location: Optional[str] = None,
credentials: Optional[auth_credentials.Credentials] = None,
encryption_spec_key_name: Optional[str] = None,
sync: Optional[bool] = True,
) -> "Featurestore":
""""""
raise NotImplementedError

def create_entity_type(
self,
entity_type_id: str,
description: Optional[str] = None,
labels: Optional[Dict[str, str]] = None,
request_metadata: Optional[Sequence[Tuple[str, str]]] = (),
sync: Optional[bool] = True,
) -> "_featurestores.EntityType":
""""""
raise NotImplementedError

def batch_serve_to_bq(
self,
entity_type_ids: List[str],
destination_feature_mappings: Dict[str, Union[List[str], Dict[str, str]]],
read_instance: Optional[Union[str, List[str], pd.DataFrame]] = None,
pass_through_field: Optional[List[str]] = None,
bq_destination_output_uri: Optional[str] = None,
sync: Optional[bool] = True,
) -> None:
""""""
raise NotImplementedError

def batch_serve_to_gcs(
self,
entity_type_ids: List[str],
destination_feature_mappings: Dict[str, Union[List[str], Dict[str, str]]],
read_instance: Optional[Union[str, List[str], pd.DataFrame]] = None,
pass_through_field: Optional[List[str]] = None,
gcs_destination_output_uri_prefix: Optional[str] = None,
gcs_destination_type: Optional[str] = None,
sync: Optional[bool] = True,
) -> None:
""""""
raise NotImplementedError

def batch_serve_to_df(
self,
entity_type_ids: List[str],
destination_feature_mappings: Dict[str, Union[List[str], Dict[str, str]]],
read_instance: Optional[Union[str, List[str], pd.DataFrame]] = None,
pass_through_field: Optional[List[str]] = None,
) -> pd.DataFrame:
""""""
raise NotImplementedError

0 comments on commit 4b35fcd

Please sign in to comment.