snowflake-ml-python 1.8.2__py3-none-any.whl → 1.8.4__py3-none-any.whl

snowflake/ml/data/torch_utils.py

@@ -1,4 +1,4 @@
-from typing import Any, Dict, Iterator, List, Optional, Union
+from typing import Any, Iterator, Optional, Union
 
 import numpy as np
 import numpy.typing as npt
@@ -7,7 +7,7 @@ import torch.utils.data
 from snowflake.ml.data import data_ingestor
 
 
-class TorchDatasetWrapper(torch.utils.data.IterableDataset[Dict[str, Any]]):
+class TorchDatasetWrapper(torch.utils.data.IterableDataset[dict[str, Any]]):
     """Wrap a DataIngestor into a PyTorch IterableDataset"""
 
     def __init__(
@@ -32,7 +32,7 @@ class TorchDatasetWrapper(torch.utils.data.IterableDataset[Dict[str, Any]]):
         self._squeeze_outputs = squeeze
         self._expand_dims = expand_dims
 
-    def __iter__(self) -> Iterator[Dict[str, Union[npt.NDArray[Any], List[Any]]]]:
+    def __iter__(self) -> Iterator[dict[str, Union[npt.NDArray[Any], list[Any]]]]:
         max_idx = 0
         filter_idx = 0
         worker_info = torch.utils.data.get_worker_info()
@@ -59,7 +59,7 @@ class TorchDatasetWrapper(torch.utils.data.IterableDataset[Dict[str, Any]]):
                 counter = 0
 
 
-class TorchDataPipeWrapper(TorchDatasetWrapper, torch.utils.data.IterDataPipe[Dict[str, Any]]):
+class TorchDataPipeWrapper(TorchDatasetWrapper, torch.utils.data.IterDataPipe[dict[str, Any]]):
     """Wrap a DataIngestor into a PyTorch IterDataPipe"""
 
     def __init__(
@@ -77,7 +77,7 @@ class TorchDataPipeWrapper(TorchDatasetWrapper, torch.utils.data.IterDataPipe[Di
 
 def _preprocess_array(
     arr: npt.NDArray[Any], squeeze: bool = False, expand_dims: bool = True
-) -> Union[npt.NDArray[Any], List[np.object_]]:
+) -> Union[npt.NDArray[Any], list[np.object_]]:
     """Preprocesses batch column values."""
     single_dimensional = arr.ndim < 2 and not arr.dtype == np.object_
 

snowflake/ml/dataset/dataset.py

@@ -1,7 +1,7 @@
 import json
 import warnings
 from datetime import datetime
-from typing import Any, Dict, List, Optional, Tuple, Union
+from typing import Any, Optional, Union
 
 from snowflake import snowpark
 from snowflake.ml._internal import telemetry
@@ -46,8 +46,8 @@ class DatasetVersion:
         self._version = version
         self._session: snowpark.Session = self._parent._session
 
-        self._properties: Optional[Dict[str, Any]] = None
-        self._raw_metadata: Optional[Dict[str, Any]] = None
+        self._properties: Optional[dict[str, Any]] = None
+        self._raw_metadata: Optional[dict[str, Any]] = None
         self._metadata: Optional[dataset_metadata.DatasetMetadata] = None
 
     @property
@@ -66,14 +66,14 @@ class DatasetVersion:
         return comment
 
     @property
-    def label_cols(self) -> List[str]:
+    def label_cols(self) -> list[str]:
         metadata = self._get_metadata()
         if metadata is None or metadata.label_cols is None:
             return []
         return metadata.label_cols
 
     @property
-    def exclude_cols(self) -> List[str]:
+    def exclude_cols(self) -> list[str]:
         metadata = self._get_metadata()
         if metadata is None or metadata.exclude_cols is None:
             return []
@@ -115,7 +115,7 @@ class DatasetVersion:
         return path
 
     @telemetry.send_api_usage_telemetry(project=_PROJECT)
-    def list_files(self, subdir: Optional[str] = None) -> List[snowpark.Row]:
+    def list_files(self, subdir: Optional[str] = None) -> list[snowpark.Row]:
         """Get the list of remote file paths for the current DatasetVersion."""
         return self._session.sql(f"LIST {self.url()}{subdir or ''}").collect(
             statement_params=_TELEMETRY_STATEMENT_PARAMS
@@ -244,7 +244,7 @@ class Dataset(lineage_node.LineageNode):
                 raise
 
     @telemetry.send_api_usage_telemetry(project=_PROJECT)
-    def list_versions(self, detailed: bool = False) -> Union[List[str], List[snowpark.Row]]:
+    def list_versions(self, detailed: bool = False) -> Union[list[str], list[snowpark.Row]]:
         """Return list of versions"""
         versions = self._list_versions()
         versions.sort(key=lambda r: r[_DATASET_VERSION_NAME_COL])
@@ -271,8 +271,8 @@ class Dataset(lineage_node.LineageNode):
         version: str,
         input_dataframe: snowpark.DataFrame,
         shuffle: bool = False,
-        exclude_cols: Optional[List[str]] = None,
-        label_cols: Optional[List[str]] = None,
+        exclude_cols: Optional[list[str]] = None,
+        label_cols: Optional[list[str]] = None,
         properties: Optional[dataset_metadata.DatasetPropertiesType] = None,
         partition_by: Optional[str] = None,
         comment: Optional[str] = None,
@@ -423,7 +423,7 @@ class Dataset(lineage_node.LineageNode):
             statement_params=_TELEMETRY_STATEMENT_PARAMS
         )
 
-    def _list_versions(self, pattern: Optional[str] = None) -> List[snowpark.Row]:
+    def _list_versions(self, pattern: Optional[str] = None) -> list[snowpark.Row]:
         """Return list of versions"""
         try:
             pattern_clause = f" LIKE '{pattern}'" if pattern else ""
@@ -469,7 +469,7 @@ lineage_node.DOMAIN_LINEAGE_REGISTRY["dataset"] = Dataset
 # Utility methods
 
 
-def _get_schema_level_identifier(session: snowpark.Session, dataset_name: str) -> Tuple[str, str, str]:
+def _get_schema_level_identifier(session: snowpark.Session, dataset_name: str) -> tuple[str, str, str]:
     """Resolve a dataset name into a validated schema-level location identifier"""
     db, schema, object_name = identifier.parse_schema_level_object_identifier(dataset_name)
     db = db or session.get_current_database()

snowflake/ml/dataset/dataset_metadata.py

@@ -1,7 +1,7 @@
 import dataclasses
 import json
 import typing
-from typing import Any, Dict, List, Optional, Union
+from typing import Any, Optional, Union
 
 _PROPERTY_TYPE_KEY = "$proptype$"
 DATASET_SCHEMA_VERSION = "1"
@@ -20,15 +20,15 @@ class FeatureStoreMetadata:
     """
 
     spine_query: str
-    serialized_feature_views: Optional[List[str]] = None
-    compact_feature_views: Optional[List[str]] = None
+    serialized_feature_views: Optional[list[str]] = None
+    compact_feature_views: Optional[list[str]] = None
     spine_timestamp_col: Optional[str] = None
 
     def to_json(self) -> str:
         return json.dumps(dataclasses.asdict(self))
 
     @classmethod
-    def from_json(cls, input_json: Union[Dict[str, Any], str, bytes]) -> "FeatureStoreMetadata":
+    def from_json(cls, input_json: Union[dict[str, Any], str, bytes]) -> "FeatureStoreMetadata":
         if isinstance(input_json, dict):
             return cls(**input_json)
         return cls(**json.loads(input_json))
@@ -61,8 +61,8 @@ class DatasetMetadata:
 
     source_query: str
     owner: str
-    exclude_cols: Optional[List[str]] = None
-    label_cols: Optional[List[str]] = None
+    exclude_cols: Optional[list[str]] = None
+    label_cols: Optional[list[str]] = None
     properties: Optional[DatasetPropertiesType] = None
     schema_version: str = dataclasses.field(default=DATASET_SCHEMA_VERSION, init=False)
 
@@ -78,11 +78,11 @@ class DatasetMetadata:
         return json.dumps(state_dict)
 
     @classmethod
-    def from_json(cls, input_json: Union[Dict[str, Any], str, bytes]) -> "DatasetMetadata":
+    def from_json(cls, input_json: Union[dict[str, Any], str, bytes]) -> "DatasetMetadata":
         if not input_json:
             raise ValueError("json_str was empty or None")
         try:
-            state_dict: Dict[str, Any] = (
+            state_dict: dict[str, Any] = (
                 input_json if isinstance(input_json, dict) else json.loads(input_json, strict=False)
             )
 

snowflake/ml/dataset/dataset_reader.py

@@ -1,4 +1,4 @@
-from typing import Any, List, Optional, Type
+from typing import Any, Optional
 
 from snowflake import snowpark
 from snowflake.ml._internal import telemetry
@@ -25,21 +25,21 @@ class DatasetReader(data_connector.DataConnector):
 
         self._session: snowpark.Session = snowpark_session
         self._fs: snowfs.SnowFileSystem = ingestor_utils.get_dataset_filesystem(self._session)
-        self._files: Optional[List[str]] = None
+        self._files: Optional[list[str]] = None
 
     @classmethod
     def from_dataframe(
-        cls, df: snowpark.DataFrame, ingestor_class: Optional[Type[data_ingestor.DataIngestor]] = None, **kwargs: Any
+        cls, df: snowpark.DataFrame, ingestor_class: Optional[type[data_ingestor.DataIngestor]] = None, **kwargs: Any
     ) -> "DatasetReader":
         # Block superclass constructor from Snowpark DataFrames
         raise RuntimeError("Creating DatasetReader from DataFrames not supported")
 
-    def _list_files(self) -> List[str]:
+    def _list_files(self) -> list[str]:
         """Private helper function that lists all files in this DatasetVersion and caches the results."""
         if self._files:
             return self._files
 
-        files: List[str] = []
+        files: list[str] = []
         for source in self.data_sources:
             assert isinstance(source, data_source.DatasetInfo)
             files.extend(ingestor_utils.get_dataset_files(self._session, source, filesystem=self._fs))
@@ -49,7 +49,7 @@ class DatasetReader(data_connector.DataConnector):
         return self._files
 
     @telemetry.send_api_usage_telemetry(project=_PROJECT, subproject=_SUBPROJECT)
-    def files(self) -> List[str]:
+    def files(self) -> list[str]:
         """Get the list of remote file paths for the current DatasetVersion.
 
         The file paths follows the snow protocol.
@@ -92,7 +92,7 @@ class DatasetReader(data_connector.DataConnector):
                 For example, an OBJECT column may be scanned back as a STRING column.
         """
         file_path_pattern = ".*data_.*[.]parquet"
-        dfs: List[snowpark.DataFrame] = []
+        dfs: list[snowpark.DataFrame] = []
         for source in self.data_sources:
             assert isinstance(source, data_source.DatasetInfo) and source.url is not None
             stage_reader = self._session.read.option("pattern", file_path_pattern)

snowflake/ml/feature_store/__init__.py

@@ -4,7 +4,7 @@ from snowflake.ml._internal import init_utils
 
 from .access_manager import setup_feature_store
 
-pkg_dir = os.path.dirname(os.path.abspath(__file__))
+pkg_dir = os.path.dirname(__file__)
 pkg_name = __name__
 exportable_classes = init_utils.fetch_classes_from_modules_in_pkg_dir(pkg_dir=pkg_dir, pkg_name=pkg_name)
 for k, v in exportable_classes.items():

snowflake/ml/feature_store/access_manager.py

@@ -1,6 +1,6 @@
 from dataclasses import asdict, dataclass
 from enum import Enum
-from typing import Dict, List, Optional
+from typing import Optional
 from warnings import warn
 
 from snowflake.ml._internal import telemetry
@@ -28,7 +28,7 @@ class _FeatureStoreRole(Enum):
 class _Privilege:
     object_type: str
     object_name: str
-    privileges: List[str]
+    privileges: list[str]
     scope: Optional[str] = None
     optional: bool = False
 
@@ -41,7 +41,7 @@ class _SessionInfo:
 
 
 # Lists of permissions as tuples of (OBJECT_TYPE, [PRIVILEGES, ...])
-_PRE_INIT_PRIVILEGES: Dict[_FeatureStoreRole, List[_Privilege]] = {
+_PRE_INIT_PRIVILEGES: dict[_FeatureStoreRole, list[_Privilege]] = {
     _FeatureStoreRole.PRODUCER: [
         _Privilege("DATABASE", "{database}", ["USAGE"]),
         _Privilege("SCHEMA", "{database}.{schema}", ["USAGE"]),
@@ -78,7 +78,7 @@ _PRE_INIT_PRIVILEGES: Dict[_FeatureStoreRole, List[_Privilege]] = {
     _FeatureStoreRole.NONE: [],
 }
 
-_POST_INIT_PRIVILEGES: Dict[_FeatureStoreRole, List[_Privilege]] = {
+_POST_INIT_PRIVILEGES: dict[_FeatureStoreRole, list[_Privilege]] = {
     _FeatureStoreRole.PRODUCER: [
         _Privilege("TAG", f"{{database}}.{{schema}}.{_FEATURE_VIEW_METADATA_TAG}", ["APPLY"]),
         _Privilege("TAG", f"{{database}}.{{schema}}.{_FEATURE_STORE_OBJECT_TAG}", ["APPLY"]),
@@ -89,7 +89,7 @@ _POST_INIT_PRIVILEGES: Dict[_FeatureStoreRole, List[_Privilege]] = {
 
 
 def _grant_privileges(
-    session: Session, role_name: str, privileges: List[_Privilege], session_info: _SessionInfo
+    session: Session, role_name: str, privileges: list[_Privilege], session_info: _SessionInfo
 ) -> None:
     session_info_dict = asdict(session_info)
     for p in privileges:
@@ -129,7 +129,7 @@ def _grant_privileges(
 def _configure_pre_init_privileges(
     session: Session,
     session_info: _SessionInfo,
-    roles_to_create: Dict[_FeatureStoreRole, str],
+    roles_to_create: dict[_FeatureStoreRole, str],
 ) -> None:
     """
     Configure Feature Store role privileges. Must be run with ACCOUNTADMIN
@@ -172,7 +172,7 @@ def _configure_pre_init_privileges(
 def _configure_post_init_privileges(
     session: Session,
     session_info: _SessionInfo,
-    roles_to_create: Dict[_FeatureStoreRole, str],
+    roles_to_create: dict[_FeatureStoreRole, str],
 ) -> None:
     for role_type, role in roles_to_create.items():
         _grant_privileges(session, role, _POST_INIT_PRIVILEGES[role_type], session_info)

snowflake/ml/feature_store/entity.py

@@ -1,4 +1,4 @@
-from typing import Dict, List, Optional
+from typing import Optional
 
 from snowflake.ml._internal.utils.sql_identifier import (
     SqlIdentifier,
@@ -22,7 +22,7 @@ class Entity:
     It can also be used for FeatureView search and lineage tracking.
     """
 
-    def __init__(self, name: str, join_keys: List[str], *, desc: str = "") -> None:
+    def __init__(self, name: str, join_keys: list[str], *, desc: str = "") -> None:
         """
         Creates an Entity instance.
 
@@ -51,11 +51,11 @@ class Entity:
         self._validate(name, join_keys)
 
         self.name: SqlIdentifier = SqlIdentifier(name)
-        self.join_keys: List[SqlIdentifier] = to_sql_identifiers(join_keys)
+        self.join_keys: list[SqlIdentifier] = to_sql_identifiers(join_keys)
         self.owner: Optional[str] = None
         self.desc: str = desc
 
-    def _validate(self, name: str, join_keys: List[str]) -> None:
+    def _validate(self, name: str, join_keys: list[str]) -> None:
         if len(name) > _ENTITY_NAME_LENGTH_LIMIT:
             raise ValueError(f"Entity name `{name}` exceeds maximum length: {_ENTITY_NAME_LENGTH_LIMIT}")
         if _FEATURE_VIEW_ENTITY_TAG_DELIMITER in name:
@@ -73,7 +73,7 @@ class Entity:
             if len(k) > _ENTITY_JOIN_KEY_LENGTH_LIMIT:
                 raise ValueError(f"Join key: {k} exceeds length limit {_ENTITY_JOIN_KEY_LENGTH_LIMIT}.")
 
-    def _to_dict(self) -> Dict[str, str]:
+    def _to_dict(self) -> dict[str, str]:
         entity_dict = self.__dict__.copy()
         for k, v in entity_dict.items():
             if isinstance(v, SqlIdentifier):
@@ -81,7 +81,7 @@ class Entity:
         return entity_dict
 
     @staticmethod
-    def _construct_entity(name: str, join_keys: List[str], desc: str, owner: str) -> "Entity":
+    def _construct_entity(name: str, join_keys: list[str], desc: str, owner: str) -> "Entity":
         e = Entity(name, join_keys, desc=desc)
         e.owner = owner
         return e

snowflake/ml/feature_store/examples/airline_features/entities.py

@@ -1,5 +1,3 @@
-from typing import List
-
 from snowflake.ml.feature_store import Entity
 
 zipcode_entity = Entity(
@@ -12,5 +10,5 @@ plane_entity = Entity(name="PLANE_MODEL", join_keys=["PLANE_MODEL"], desc="The m
 
 
 # This will be invoked by example_helper.py. Do not change function name.
-def get_all_entities() -> List[Entity]:
+def get_all_entities() -> list[Entity]:
     return [zipcode_entity, plane_entity]

snowflake/ml/feature_store/examples/airline_features/features/plane_features.py

@@ -1,5 +1,3 @@
-from typing import List
-
 from snowflake.ml.feature_store import FeatureView
 from snowflake.ml.feature_store.examples.airline_features.entities import plane_entity
 from snowflake.snowpark import DataFrame, Session
@@ -7,7 +5,7 @@ from snowflake.snowpark import DataFrame, Session
 
 # This function will be invoked by example_helper.py. Do not change the name.
 def create_draft_feature_view(
-    session: Session, source_dfs: List[DataFrame], source_tables: List[str], database: str, schema: str
+    session: Session, source_dfs: list[DataFrame], source_tables: list[str], database: str, schema: str
 ) -> FeatureView:
     """Create a feature view about airplane model."""
     query = session.sql(

snowflake/ml/feature_store/examples/airline_features/features/weather_features.py

@@ -1,5 +1,3 @@
-from typing import List
-
 from snowflake.ml.feature_store import FeatureView
 from snowflake.ml.feature_store.examples.airline_features.entities import zipcode_entity
 from snowflake.snowpark import DataFrame, Session
@@ -7,7 +5,7 @@ from snowflake.snowpark import DataFrame, Session
 
 # This function will be invoked by example_helper.py. Do not change the name.
 def create_draft_feature_view(
-    session: Session, source_dfs: List[DataFrame], source_tables: List[str], database: str, schema: str
+    session: Session, source_dfs: list[DataFrame], source_tables: list[str], database: str, schema: str
 ) -> FeatureView:
     """Create a feature view about airport weather."""
     query = session.sql(

snowflake/ml/feature_store/examples/citibike_trip_features/entities.py

@@ -1,5 +1,3 @@
-from typing import List
-
 from snowflake.ml.feature_store import Entity
 
 end_station_id = Entity(
@@ -16,5 +14,5 @@ trip_id = Entity(
 
 
 # This will be invoked by example_helper.py. Do not change function name.
-def get_all_entities() -> List[Entity]:
+def get_all_entities() -> list[Entity]:
     return [end_station_id, trip_id]

snowflake/ml/feature_store/examples/citibike_trip_features/features/station_feature.py

@@ -1,5 +1,3 @@
-from typing import List
-
 from snowflake.ml.feature_store import FeatureView
 from snowflake.ml.feature_store.examples.citibike_trip_features.entities import (
     end_station_id,
@@ -9,7 +7,7 @@ from snowflake.snowpark import DataFrame, Session
 
 # This function will be invoked by example_helper.py. Do not change the name.
 def create_draft_feature_view(
-    session: Session, source_dfs: List[DataFrame], source_tables: List[str], database: str, schema: str
+    session: Session, source_dfs: list[DataFrame], source_tables: list[str], database: str, schema: str
 ) -> FeatureView:
     """Create a feature view about trip station."""
     query = session.sql(

snowflake/ml/feature_store/examples/citibike_trip_features/features/trip_feature.py

@@ -1,5 +1,3 @@
-from typing import List
-
 from snowflake.ml.feature_store import FeatureView
 from snowflake.ml.feature_store.examples.citibike_trip_features.entities import trip_id
 from snowflake.snowpark import DataFrame, Session, functions as F
@@ -7,7 +5,7 @@ from snowflake.snowpark import DataFrame, Session, functions as F
 
 # This function will be invoked by example_helper.py. Do not change the name.
 def create_draft_feature_view(
-    session: Session, source_dfs: List[DataFrame], source_tables: List[str], database: str, schema: str
+    session: Session, source_dfs: list[DataFrame], source_tables: list[str], database: str, schema: str
 ) -> FeatureView:
     """Create a feature view about trip."""
     feature_df = source_dfs[0].select(

snowflake/ml/feature_store/examples/example_helper.py

@@ -3,7 +3,7 @@ import logging
 import os
 import re
 from pathlib import Path
-from typing import Any, Dict, List, Optional
+from typing import Any, Optional
 
 import yaml
 
@@ -32,12 +32,12 @@ class ExampleHelper:
 
     def _clear(self) -> None:
         self._selected_example = None
-        self._source_tables: List[str] = []
-        self._source_dfs: List[DataFrame] = []
-        self._excluded_columns: List[sql_identifier.SqlIdentifier] = []
-        self._label_columns: List[sql_identifier.SqlIdentifier] = []
+        self._source_tables: list[str] = []
+        self._source_dfs: list[DataFrame] = []
+        self._excluded_columns: list[sql_identifier.SqlIdentifier] = []
+        self._label_columns: list[sql_identifier.SqlIdentifier] = []
         self._timestamp_column: Optional[sql_identifier.SqlIdentifier] = None
-        self._epoch_to_timestamp_cols: List[str] = []
+        self._epoch_to_timestamp_cols: list[str] = []
         self._add_id_column: Optional[sql_identifier.SqlIdentifier] = None
         self._training_spine_table: str = ""
 
@@ -53,7 +53,7 @@ class ExampleHelper:
                 rows.append((f_name, source_dict["model_category"], source_dict["desc"], source_dict["label_columns"]))
         return self._session.create_dataframe(rows, schema=["NAME", "MODEL_CATEGORY", "DESC", "LABEL_COLS"])
 
-    def load_draft_feature_views(self) -> List[FeatureView]:
+    def load_draft_feature_views(self) -> list[FeatureView]:
         """Return all feature views in an example.
 
         Returns:
@@ -74,7 +74,7 @@ class ExampleHelper:
 
         return fvs
 
-    def load_entities(self) -> List[Entity]:
+    def load_entities(self) -> list[Entity]:
         """Return all entities in an example.
 
         Returns:
@@ -88,7 +88,7 @@ class ExampleHelper:
         with open(file_path) as fs:
             return yaml.safe_load(fs)
 
-    def _create_file_format(self, format_dict: Dict[str, str], format_name: str) -> None:
+    def _create_file_format(self, format_dict: dict[str, str], format_name: str) -> None:
         """Create a file name with given name."""
         self._session.sql(
             f"""
@@ -110,7 +110,7 @@ class ExampleHelper:
             """
         ).collect()
 
-    def _load_csv(self, schema_dict: Dict[str, str], temp_stage_name: str) -> List[str]:
+    def _load_csv(self, schema_dict: dict[str, str], temp_stage_name: str) -> list[str]:
         # create temp file format
         file_format_name = f"{self._database_name}.{self._dataset_schema}.feature_store_temp_format"
         format_str = ""
@@ -145,7 +145,7 @@ class ExampleHelper:
 
         return [schema_dict["destination_table_name"]]
 
-    def _load_parquet(self, schema_dict: Dict[str, str], temp_stage_name: str) -> List[str]:
+    def _load_parquet(self, schema_dict: dict[str, str], temp_stage_name: str) -> list[str]:
         regex_pattern = schema_dict["load_files_pattern"]
         all_files = self._session.sql(f"list @{temp_stage_name}").collect()
         filtered_files = [item["name"] for item in all_files if re.match(regex_pattern, item["name"])]
@@ -187,7 +187,7 @@ class ExampleHelper:
 
         return result
 
-    def _load_source_data(self, schema_yaml_file: str) -> List[str]:
+    def _load_source_data(self, schema_yaml_file: str) -> list[str]:
         """Parse a yaml schema file and load data into Snowflake.
 
         Args:
@@ -210,7 +210,7 @@ class ExampleHelper:
         else:
             return self._load_csv(schema_dict, temp_stage_name)
 
-    def load_example(self, example_name: str) -> List[str]:
+    def load_example(self, example_name: str) -> list[str]:
         """Select the active example and load its datasets to Snowflake.
 
         Args:
@@ -247,7 +247,7 @@ class ExampleHelper:
 
         return self.load_source_data(source_yaml_data)
 
-    def load_source_data(self, source_data_name: str) -> List[str]:
+    def load_source_data(self, source_data_name: str) -> list[str]:
         """Load source data into Snowflake.
 
         Args:
@@ -269,10 +269,10 @@ class ExampleHelper:
     def get_current_schema(self) -> str:
         return self._dataset_schema
 
-    def get_label_cols(self) -> List[str]:
+    def get_label_cols(self) -> list[str]:
         return [item.resolved() for item in self._label_columns]
 
-    def get_excluded_cols(self) -> List[str]:
+    def get_excluded_cols(self) -> list[str]:
         return [item.resolved() for item in self._excluded_columns]
 
     def get_training_data_timestamp_col(self) -> Optional[str]:

snowflake/ml/feature_store/examples/new_york_taxi_features/entities.py

@@ -1,5 +1,3 @@
-from typing import List
-
 from snowflake.ml.feature_store import Entity
 
 trip_id = Entity(name="TRIP_ID", join_keys=["TRIP_ID"], desc="Trip id.")
@@ -8,5 +6,5 @@ location_id = Entity(name="DOLOCATIONID", join_keys=["DOLOCATIONID"], desc="Drop
 
 
 # This will be invoked by example_helper.py. Do not change function name.
-def get_all_entities() -> List[Entity]:
+def get_all_entities() -> list[Entity]:
     return [trip_id, location_id]

snowflake/ml/feature_store/examples/new_york_taxi_features/features/location_features.py

@@ -1,5 +1,3 @@
-from typing import List
-
 from snowflake.ml.feature_store import FeatureView
 from snowflake.ml.feature_store.examples.new_york_taxi_features.entities import (
     location_id,
@@ -9,7 +7,7 @@ from snowflake.snowpark import DataFrame, Session
 
 # This function will be invoked by example_helper.py. Do not change the name.
 def create_draft_feature_view(
-    session: Session, source_dfs: List[DataFrame], source_tables: List[str], database: str, schema: str
+    session: Session, source_dfs: list[DataFrame], source_tables: list[str], database: str, schema: str
 ) -> FeatureView:
     """Create a draft feature view."""
     feature_df = session.sql(

snowflake/ml/feature_store/examples/new_york_taxi_features/features/trip_features.py

@@ -1,5 +1,3 @@
-from typing import List
-
 from snowflake.ml.feature_store import FeatureView
 from snowflake.ml.feature_store.examples.new_york_taxi_features.entities import trip_id
 from snowflake.snowpark import DataFrame, Session
@@ -7,7 +5,7 @@ from snowflake.snowpark import DataFrame, Session
 
 # This function will be invoked by example_helper.py. Do not change the name.
 def create_draft_feature_view(
-    session: Session, source_dfs: List[DataFrame], source_tables: List[str], database: str, schema: str
+    session: Session, source_dfs: list[DataFrame], source_tables: list[str], database: str, schema: str
 ) -> FeatureView:
     """Create a draft feature view."""
     feature_df = session.sql(

snowflake/ml/feature_store/examples/wine_quality_features/entities.py

@@ -1,5 +1,3 @@
-from typing import List
-
 from snowflake.ml.feature_store import Entity
 
 wine_id = Entity(
@@ -10,5 +8,5 @@ wine_id = Entity(
 
 
 # This will be invoked by example_helper.py. Do not change function name.
-def get_all_entities() -> List[Entity]:
+def get_all_entities() -> list[Entity]:
     return [wine_id]

snowflake/ml/feature_store/examples/wine_quality_features/features/managed_wine_features.py

@@ -1,5 +1,3 @@
-from typing import List
-
 from snowflake.ml.feature_store import FeatureView
 from snowflake.ml.feature_store.examples.wine_quality_features.entities import wine_id
 from snowflake.snowpark import DataFrame, Session, functions as F
@@ -7,7 +5,7 @@ from snowflake.snowpark import DataFrame, Session, functions as F
 
 # This function will be invoked by example_helper.py. Do not change the name.
 def create_draft_feature_view(
-    session: Session, source_dfs: List[DataFrame], source_tables: List[str], database: str, schema: str
+    session: Session, source_dfs: list[DataFrame], source_tables: list[str], database: str, schema: str
 ) -> FeatureView:
     """Create a feature view about trip station."""
     feature_df = source_dfs[0].select(

snowflake/ml/feature_store/examples/wine_quality_features/features/static_wine_features.py

@@ -1,5 +1,3 @@
-from typing import List
-
 from snowflake.ml.feature_store import FeatureView
 from snowflake.ml.feature_store.examples.wine_quality_features.entities import wine_id
 from snowflake.snowpark import DataFrame, Session
@@ -7,7 +5,7 @@ from snowflake.snowpark import DataFrame, Session
 
 # This function will be invoked by example_helper.py. Do not change the name.
 def create_draft_feature_view(
-    session: Session, source_dfs: List[DataFrame], source_tables: List[str], database: str, schema: str
+    session: Session, source_dfs: list[DataFrame], source_tables: list[str], database: str, schema: str
 ) -> FeatureView:
     """Create a feature view about trip station."""
     feature_df = source_dfs[0].select("WINE_ID", "SULPHATES", "ALCOHOL")