frogml-core 0.0.111__py3-none-any.whl → 0.0.112__py3-none-any.whl

frogml_core/__init__.py

@@ -1,7 +1,7 @@
 """Top-level package for frogml."""
 
 __author__ = "jfrog"
-__version__ = "0.0.111"
+__version__ = "0.0.112"
 
 from frogml_core.inner.di_configuration import wire_dependencies
 from frogml_core.model.model_version_tracking import (  # noqa: F401,E501

frogml_core/automations/automations.py

@@ -293,6 +293,7 @@ class Automation:
     create_audit: AutomationAudit = field(default_factory=AutomationAudit)
     on_error: Notification = field(default=None)
     on_success: Notification = field(default=None)
+    jfrog_token_id: str = field(default="")
 
     def to_proto(self):
         return AutomationProto(
@@ -315,10 +316,11 @@ class Automation:
             qwak_environment_id=self.environment,
             create_audit=self.create_audit.to_proto(),
             is_deleted=self.is_deleted,
+            jfrog_token_id=self.jfrog_token_id,
         )
 
-    @staticmethod
-    def from_proto(message: AutomationProto):
+    @classmethod
+    def from_proto(cls, message: AutomationProto):
         action = map_action_name_to_class(
             message.automation_spec.action.WhichOneof("action")
         )
@@ -332,7 +334,7 @@ class Automation:
             message.automation_spec.on_success.WhichOneof("notification")
         )
 
-        return Automation(
+        return cls(
             id=message.automation_id,
             name=message.automation_spec.automation_name,
             description=message.automation_spec.automation_description,
@@ -359,6 +361,7 @@ class Automation:
                 if on_success_notification
                 else None
             ),
+            jfrog_token_id=message.jfrog_token_id,
         )
 
     def __str__(self):

frogml_core/feature_store/data_sources/batch/_jdbc.py

@@ -13,6 +13,7 @@ class JdbcSource(BaseBatchSource, ABC):
     url: Optional[str] = None
     db_table: Optional[str] = None
     query: Optional[str] = None
+    repository: Optional[str] = None
 
     def __post_init__(self):
         self._validate()

frogml_core/feature_store/data_sources/batch/athena.py

@@ -52,6 +52,7 @@ class AthenaSource(JdbcSource):
     aws_authentication: Optional[AwsAuthentication] = None
     time_partition_columns: Optional[TimePartitionColumns] = None
     workgroup: Optional[str] = None
+    repository: Optional[str] = None
 
     def _validate(self):
         super()._validate()
@@ -67,7 +68,7 @@ class AthenaSource(JdbcSource):
                 "At least one of: `s3_output_location`, `workgroup` must be set"
             )
 
-    def _to_proto(self, artifact_url: Optional[str] = None):
+    def _to_proto(self, artifact_url: Optional[str] = None) -> ProtoDataSourceSpec:
         proto_date_partition_columns = None
         proto_time_fragment_partition_columns = None
 
@@ -97,6 +98,7 @@ class AthenaSource(JdbcSource):
         )
 
         return ProtoDataSourceSpec(
+            data_source_repository_name=self.repository,
             batch_source=ProtoBatchSource(
                 name=self.name,
                 description=self.description,
@@ -116,7 +118,7 @@ class AthenaSource(JdbcSource):
                         workgroup=self.workgroup,
                     ),
                 ),
-            )
+            ),
         )
 
     @classmethod

frogml_core/feature_store/data_sources/batch/big_query.py

@@ -25,6 +25,7 @@ class BigQuerySource(BaseBatchSource):
     views_enabled: Optional[bool] = False
     materialization_dataset: Optional[str] = None
     materialization_project: Optional[str] = None
+    repository: Optional[str] = None
 
     def __post_init__(self):
         if not (self.sql or (self.table and self.dataset)):
@@ -59,8 +60,9 @@ class BigQuerySource(BaseBatchSource):
             materialization_dataset=bigquery.materialization_dataset,
         )
 
-    def _to_proto(self, artifact_url: Optional[str] = None):
+    def _to_proto(self, artifact_url: Optional[str] = None) -> ProtoDataSourceSpec:
         return ProtoDataSourceSpec(
+            data_source_repository_name=self.repository,
             batch_source=ProtoBatchSource(
                 name=self.name,
                 description=self.description,
@@ -76,5 +78,5 @@ class BigQuerySource(BaseBatchSource):
                     materialization_dataset=self.materialization_dataset,
                     materialization_project=self.materialization_project,
                 ),
-            )
+            ),
         )

frogml_core/feature_store/data_sources/batch/clickhouse.py

@@ -21,6 +21,7 @@ class ClickhouseSource(BaseBatchSource):
     url: Optional[str] = None
     db_table: Optional[str] = None
     query: Optional[str] = None
+    repository: Optional[str] = None
 
     def __post_init__(self):
         self._validate()
@@ -29,8 +30,9 @@ class ClickhouseSource(BaseBatchSource):
         if not (bool(self.db_table) ^ bool(self.query)):
             raise FrogmlException("Only one of query and db_table must be set")
 
-    def _to_proto(self, artifact_url: Optional[str] = None):
+    def _to_proto(self, artifact_url: Optional[str] = None) -> ProtoDataSourceSpec:
         return ProtoDataSourceSpec(
+            data_source_repository_name=self.repository,
             batch_source=ProtoBatchSource(
                 name=self.name,
                 description=self.description,
@@ -42,7 +44,7 @@ class ClickhouseSource(BaseBatchSource):
                     table=self.db_table,
                     sql=self.query,
                 ),
-            )
+            ),
         )
 
     @classmethod

frogml_core/feature_store/data_sources/batch/csv.py

@@ -25,6 +25,7 @@ class CsvSource(BaseBatchSource):
     quote_character: str = '"'
     escape_character: str = '"'
     filesystem_configuration: Optional[FileSystemConfiguration] = None
+    repository: Optional[str] = None
 
     @classmethod
     def _from_proto(cls, proto):
@@ -42,12 +43,13 @@ class CsvSource(BaseBatchSource):
             filesystem_configuration=fs_conf,
         )
 
-    def _to_proto(self, artifact_url: Optional[str] = None):
+    def _to_proto(self, artifact_url: Optional[str] = None) -> ProtoDataSourceSpec:
         fs_conf = None
         if self.filesystem_configuration:
             fs_conf = self.filesystem_configuration._to_proto()
 
         return ProtoDataSourceSpec(
+            featureset_repository_name=self.repository,
             batch_source=ProtoBatchSource(
                 name=self.name,
                 description=self.description,
@@ -58,5 +60,5 @@ class CsvSource(BaseBatchSource):
                     escape_character=self.escape_character,
                     filesystem_configuration=fs_conf,
                 ),
-            )
+            ),
         )

frogml_core/feature_store/data_sources/batch/elastic_search.py

@@ -23,6 +23,7 @@ class ElasticSearchSource(BaseBatchSource):
     password_secret_name: Optional[str] = None
     exclude_fields: Optional[str] = None
     parse_dates: Optional[bool] = True
+    repository: Optional[str] = None
 
     @classmethod
     def _from_proto(cls, proto):
@@ -41,8 +42,9 @@ class ElasticSearchSource(BaseBatchSource):
             parse_dates=elastic.parse_dates,
         )
 
-    def _to_proto(self, artifact_url: Optional[str] = None):
+    def _to_proto(self, artifact_url: Optional[str] = None) -> ProtoDataSourceSpec:
         return ProtoDataSourceSpec(
+            data_source_repository_name=self.repository,
             batch_source=ProtoBatchSource(
                 name=self.name,
                 description=self.description,
@@ -57,5 +59,5 @@ class ElasticSearchSource(BaseBatchSource):
                     exclude_fields=self.exclude_fields,
                     parse_dates=self.parse_dates,
                 ),
-            )
+            ),
         )

frogml_core/feature_store/data_sources/batch/mongodb.py

@@ -22,6 +22,7 @@ class MongoDbSource(BaseBatchSource):
     collection: str
     connection_params: str
     protocol: str = "mongodb"
+    repository: Optional[str] = None
 
     @classmethod
     def _from_proto(cls, proto):
@@ -39,8 +40,9 @@ class MongoDbSource(BaseBatchSource):
             protocol=mongo.protocol,
         )
 
-    def _to_proto(self, artifact_url: Optional[str] = None):
+    def _to_proto(self, artifact_url: Optional[str] = None) -> ProtoBatchSource:
         return ProtoDataSourceSpec(
+            data_source_repository_name=self.repository,
             batch_source=ProtoBatchSource(
                 name=self.name,
                 description=self.description,
@@ -54,5 +56,5 @@ class MongoDbSource(BaseBatchSource):
                     connection_params=self.connection_params,
                     protocol=self.protocol,
                 ),
-            )
+            ),
         )

frogml_core/feature_store/data_sources/batch/mysql.py

@@ -32,8 +32,9 @@ class MysqlSource(JdbcSource):
             query=mysql.query,
         )
 
-    def _to_proto(self, artifact_url: Optional[str] = None):
+    def _to_proto(self, artifact_url: Optional[str] = None) -> ProtoJdbcSource:
         return ProtoDataSourceSpec(
+            data_source_repository_name=self.repository,
             batch_source=ProtoBatchSource(
                 name=self.name,
                 description=self.description,
@@ -46,5 +47,5 @@ class MysqlSource(JdbcSource):
                     query=self.query,
                     mysqlSource=ProtoMysqlSource(),
                 ),
-            )
+            ),
         )

frogml_core/feature_store/data_sources/batch/parquet.py

@@ -26,6 +26,7 @@ class ParquetSource(BaseBatchSource):
     filesystem_configuration: FileSystemConfiguration = field(
         default_factory=lambda: AnonymousS3Configuration()
     )
+    repository: Optional[str] = None
 
     @classmethod
     def _from_proto(cls, proto):
@@ -40,12 +41,13 @@ class ParquetSource(BaseBatchSource):
             filesystem_configuration=fs_conf,
         )
 
-    def _to_proto(self, artifact_url: Optional[str] = None):
+    def _to_proto(self, artifact_url: Optional[str] = None) -> ProtoDataSourceSpec:
         fs_conf = None
         if self.filesystem_configuration:
             fs_conf = self.filesystem_configuration._to_proto()
 
         return ProtoDataSourceSpec(
+            data_source_repository_name=self.repository,
             batch_source=ProtoBatchSource(
                 name=self.name,
                 description=self.description,
@@ -53,5 +55,5 @@ class ParquetSource(BaseBatchSource):
                 parquetSource=ProtoParquetSource(
                     path=self.path, filesystem_configuration=fs_conf
                 ),
-            )
+            ),
         )

frogml_core/feature_store/data_sources/batch/postgres.py

@@ -18,8 +18,9 @@ from frogml_core.feature_store.data_sources.batch._jdbc import JdbcSource
 
 @dataclass
 class PostgresSource(JdbcSource):
-    def _to_proto(self, artifact_url: Optional[str] = None):
+    def _to_proto(self, artifact_url: Optional[str] = None) -> ProtoDataSourceSpec:
         return ProtoDataSourceSpec(
+            data_source_repository_name=self.repository,
             batch_source=ProtoBatchSource(
                 name=self.name,
                 description=self.description,
@@ -32,7 +33,7 @@ class PostgresSource(JdbcSource):
                     query=self.query,
                     postgresqlSource=ProtoPostgresqlSource(),
                 ),
-            )
+            ),
         )
 
     @classmethod

frogml_core/feature_store/data_sources/batch/redshift.py

@@ -27,6 +27,8 @@ class RedshiftSource(JdbcSource):
     secret_access_key: Optional[str] = None
     query_group: Optional[str] = "_qwak_featurestore"
 
+    repository: Optional[str] = None
+
     def _validate(self):
         authentication_methods = sum(
             [
@@ -46,8 +48,9 @@ class RedshiftSource(JdbcSource):
                 "or user id secret name and password secret name"
             )
 
-    def _to_proto(self, artifact_url: Optional[str] = None):
+    def _to_proto(self, artifact_url: Optional[str] = None) -> ProtoDataSourceSpec:
         return ProtoDataSourceSpec(
+            data_source_repository_name=self.repository,
             batch_source=ProtoBatchSource(
                 name=self.name,
                 description=self.description,
@@ -65,7 +68,7 @@ class RedshiftSource(JdbcSource):
                         secret_access_key=self.secret_access_key,
                     ),
                 ),
-            )
+            ),
         )
 
     @classmethod

frogml_core/feature_store/data_sources/batch/snowflake.py

@@ -24,6 +24,7 @@ class SnowflakeSource(BaseBatchSource):
     warehouse: str
     table: Optional[str] = None
     query: Optional[str] = None
+    repository: Optional[str] = None
 
     def __post_init__(self):
         self._validate()
@@ -40,8 +41,9 @@ class SnowflakeSource(BaseBatchSource):
         if not self.password_secret_name:
             raise FrogmlException("password_secret_name must be set!")
 
-    def _to_proto(self, artifact_url: Optional[str] = None):
+    def _to_proto(self, artifact_url: Optional[str] = None) -> ProtoDataSourceSpec:
         return ProtoDataSourceSpec(
+            data_source_repository_name=self.repository,
             batch_source=ProtoBatchSource(
                 name=self.name,
                 description=self.description,
@@ -56,7 +58,7 @@ class SnowflakeSource(BaseBatchSource):
                     table=self.table,
                     query=self.query,
                 ),
-            )
+            ),
         )
 
     @classmethod

frogml_core/feature_store/data_sources/batch/vertica.py

@@ -22,6 +22,7 @@ class VerticaSource(BaseBatchSource):
     table: str
     username_secret_name: str
     password_secret_name: str
+    repository: Optional[str] = None
 
     @classmethod
     def _from_proto(cls, proto):
@@ -39,8 +40,9 @@ class VerticaSource(BaseBatchSource):
             table=vertica.table,
         )
 
-    def _to_proto(self, artifact_url: Optional[str] = None):
+    def _to_proto(self, artifact_url: Optional[str] = None) -> ProtoDataSourceSpec:
         return ProtoDataSourceSpec(
+            data_source_repository_name=self.repository,
             batch_source=ProtoBatchSource(
                 name=self.name,
                 description=self.description,
@@ -54,5 +56,5 @@ class VerticaSource(BaseBatchSource):
                     port=self.port,
                     table=self.table,
                 ),
-            )
+            ),
         )

frogml_core/feature_store/data_sources/streaming/kafka/kafka.py

@@ -52,6 +52,8 @@ class KafkaSource(BaseStreamingSource):
         default_factory=lambda: SslAuthentication()
     )
 
+    repository: Optional[str] = None
+
     def __post_init__(self):
         self._validate()
 
@@ -84,6 +86,7 @@ class KafkaSource(BaseStreamingSource):
 
     def _to_proto(self, artifact_url: Optional[str] = None) -> ProtoDataSourceSpec:
         return ProtoDataSourceSpec(
+            data_source_repository_name=self.repository,
             stream_source=ProtoStreamingSource(
                 name=self.name,
                 description=self.description,
@@ -99,7 +102,7 @@ class KafkaSource(BaseStreamingSource):
                         artifact_path=artifact_url
                     ),
                 ),
-            )
+            ),
         )
 
     @classmethod

frogml_core/feature_store/feature_sets/base_feature_set.py

@@ -33,6 +33,7 @@ class BaseFeatureSet(ABC):
     data_sources: List[str]
     entity: str = str()
     key: str = str()
+    repository: Optional[str] = None
     __instance_module_path__: Optional[str] = None
 
     def _validate(self):

frogml_core/feature_store/feature_sets/batch.py

@@ -72,6 +72,7 @@ def feature_set(
     name: str = None,
     entity: Optional[str] = None,
     key: Optional[str] = None,
+    repository: Optional[str] = None,
 ):
     """
     Define a batch scheduled feature set. Default scheduling policy is every 4 hours.
@@ -116,6 +117,7 @@ def feature_set(
             name=fs_name,
             entity=entity if entity else None,
             key=key if key else None,
+            repository=repository if repository else None,
             data_sources=data_sources,
             timestamp_column_name=timestamp_column_name,
             transformation=user_transformation,
@@ -323,6 +325,7 @@ class BatchFeatureSet(BaseFeatureSet):
             data_sources=[
                 ds.data_source.name for ds in batch_v1_def.feature_set_batch_sources
             ],
+            repository=proto.featureset_repository_name,
             timestamp_column_name=batch_v1_def.timestamp_column_name,
             scheduling_policy=batch_v1_def.scheduling_policy,
             transformation=BaseTransformation._from_proto(batch_v1_def.transformation),
@@ -404,6 +407,7 @@ class BatchFeatureSet(BaseFeatureSet):
                 git_commit=git_commit,
                 features=features,
                 entity=self._get_entity_definition(feature_registry),
+                featureset_repository_name=self.repository,
                 feature_set_type=ProtoFeatureSetType(
                     batch_feature_set_v1=ProtoBatchFeatureSetV1(
                         online_sink=True,

frogml_core/feature_store/feature_sets/streaming.py

@@ -86,6 +86,7 @@ def feature_set(
     entity: Optional[str] = None,
     key: Optional[str] = None,
     auxiliary_sinks: List[BaseSink] = [],
+    repository: Optional[str] = None,
 ):
     """
     Creates a streaming feature set for the specified entity using the given streaming data sources.
@@ -140,6 +141,7 @@ def feature_set(
             name=fs_name,
             entity=entity if entity else None,
             key=key if key else None,
+            repository=repository,
             data_sources=data_sources,
             timestamp_column_name=timestamp_column_name,
             transformation=user_transformation,
@@ -340,6 +342,7 @@ class StreamingFeatureSet(BaseFeatureSet):
 
         return cls(
             name=proto.name,
+            repository=proto.featureset_repository_name,
             entity=Entity._from_proto(proto.entity).name,
             data_sources=[ds.name for ds in streaming_def.data_sources],
             timestamp_column_name=streaming_def.timestamp_column_name,
@@ -420,6 +423,7 @@ class StreamingFeatureSet(BaseFeatureSet):
                 features=features,
                 entity=self._get_entity_definition(feature_registry),
                 feature_set_type=proto_featureset_type,
+                featureset_repository_name=self.repository,
             ),
             artifact_url,
         )
@@ -429,10 +433,12 @@ class StreamingFeatureSet(BaseFeatureSet):
 
         super()._validate()
 
+        # verify offline_scheduling_policy was set
+        if not self.offline_scheduling_policy:
+            raise FrogmlException("'offline_scheduling_policy' field must be set")
+
         # verify the cron expression is valid
-        if not self.offline_scheduling_policy or not croniter.croniter.is_valid(
-            self.offline_scheduling_policy
-        ):
+        if not croniter.croniter.is_valid(self.offline_scheduling_policy):
             raise FrogmlException(
                 f"offline scheduling policy "
                 f"'{self.offline_scheduling_policy}'"

frogml_core/feature_store/online/client.py

@@ -1,5 +1,6 @@
 import logging
 import math
+from io import StringIO
 from typing import TYPE_CHECKING, List, Optional, Sequence, Tuple
 
 from frogml_proto.qwak.ecosystem.v0.ecosystem_pb2 import (
@@ -221,7 +222,7 @@ class OnlineClient:
                     request, metadata=self._metadata
                 )
                 features_df: pd.DataFrame = pd.read_json(
-                    response_df_json.pandas_df_as_json, orient="split"
+                    StringIO(response_df_json.pandas_df_as_json), orient="split"
                 )
                 results.append(
                     pd.concat(

frogml_core/model_loggers/artifact_logger.py

@@ -1,5 +1,8 @@
-import urllib.request
+import logging
+from datetime import timedelta
 from typing import Optional
+import requests
+from requests import Response
 
 from frogml_proto.qwak.builds.build_url_pb2 import BuildVersioningTagsType
 from frogml_core.clients.build_orchestrator.client import BuildOrchestratorClient
@@ -13,8 +16,14 @@ from frogml_core.inner.model_loggers_utils import (
 )
 from frogml_proto.qwak.builds.builds_orchestrator_service_pb2 import (
     GetBuildVersioningUploadURLResponse,
+    GetBuildVersioningDownloadURLResponse,
 )
 
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger(__name__)
+
+MAX_CHUNK_SIZE = 8 * 1_024
+
 
 def log_file(
     from_path: str,
@@ -80,21 +89,40 @@ def load_file(
     Returns:
         the path to the newly created data file
     """
+    logger.info(f"Loading file to {to_path}")
     if not validate_tag(tag):
         raise FrogmlException(
             "Tag should contain only letters, numbers, underscore or hyphen"
         )
 
     model_id = validate_model(model_id)
-    download_url_response = BuildOrchestratorClient().get_build_versioning_download_url(
-        build_id=build_id, model_id=model_id, tag=tag
+    download_url_response: (
+        GetBuildVersioningDownloadURLResponse
+    ) = BuildOrchestratorClient().get_build_versioning_download_url(
+        build_id=build_id,
+        model_id=model_id,
+        tag=tag,
+        tag_type=BuildVersioningTagsType.FILE_TAG_TYPE,
     )
 
     try:
-        filename, headers = urllib.request.urlretrieve(  # nosec B310
-            url=download_url_response.download_url, filename=to_path
+        response: Response = requests.get(
+            download_url_response.download_url,
+            headers=download_url_response.headers,
+            stream=True,
+            timeout=(
+                timedelta(seconds=10).total_seconds(),  # timeout to connect
+                timedelta(minutes=20).total_seconds(),  # timeout to read
+            ),
         )
+        logger.info(f"Downloading file finished with status {response.status_code}")
+        response.raise_for_status()
+
+        with open(to_path, "wb") as f:
+            for chunk in response.iter_content(chunk_size=MAX_CHUNK_SIZE):
+                if chunk:
+                    f.write(chunk)
 
-        return filename
+        return to_path
     except Exception as error:
         raise FrogmlException(f"Unable to load save artifact locally: {str(error)}")

{frogml_core-0.0.111.dist-info → frogml_core-0.0.112.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: frogml-core
-Version: 0.0.111
+Version: 0.0.112
 Summary: frogml Core contains the necessary objects and communication tools for using the Jfrog ml Platform
 License: Apache-2.0
 Keywords: mlops,ml,deployment,serving,model

{frogml_core-0.0.111.dist-info → frogml_core-0.0.112.dist-info}/RECORD

@@ -1,7 +1,7 @@
-frogml_core/__init__.py,sha256=-f3e-D_LuTHADxQ68XUpbIvlOLbc_XbCXZ2ccy2Wl4I,778
+frogml_core/__init__.py,sha256=SNMYZzIxL9iYw_75tt9Zjp8e606xQbK4AT4rRIpY-BU,778
 frogml_core/automations/__init__.py,sha256=j2gD15MN-xVWhI5rAFsDwhL0CIyICLNT0scXsKvNBkU,1547
 frogml_core/automations/automation_executions.py,sha256=xpOb9Dq8gPPGNQDJTvBBZbNz4woZDRZY0HqnLSu7pwU,3230
-frogml_core/automations/automations.py,sha256=GKEQyQMi8sxX5oZn62PaxPi0zD8IaJRjBkhczRJxHNs,13070
+frogml_core/automations/automations.py,sha256=srG8S5Z20hUsk6uehz75dmCs8mEgNCCwuCXfj3CG8hY,13210
 frogml_core/automations/batch_execution_action.py,sha256=tFGQWI16CGNfo-DZJ0n86SBkKWbNt1Q8712wF0_OgUI,13327
 frogml_core/automations/build_and_deploy_action.py,sha256=WK9CJ0oYr6u6Sy794OMjTvLohA_aHaofx0ectWN-K3o,31962
 frogml_core/automations/common.py,sha256=98YNX-jCrmQ97k1IxRqzdsnet-rohcXp4yIdVNwpyh8,2752
@@ -124,31 +124,31 @@ frogml_core/feature_store/data_sources/attributes.py,sha256=SsiPoJkqSONhK8mnn6ft
 frogml_core/feature_store/data_sources/base.py,sha256=sCJ1CzbhRX-fgsw_Y3ucUu-tKC3i93upasJfGw3EmtE,4287
 frogml_core/feature_store/data_sources/batch/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 frogml_core/feature_store/data_sources/batch/_batch.py,sha256=xnrC1SZ4OH4b7gYLRajEBFQ2U00cA3JxKT8bKNTNHLg,204
-frogml_core/feature_store/data_sources/batch/_jdbc.py,sha256=lC-m3ZypER5vLtYo4h6ghYWeXvdfGSOdTadqS-sTb58,676
-frogml_core/feature_store/data_sources/batch/athena.py,sha256=fKKdKE1B59dbCiOQQ5amJECoyETwxswkBhDLQ-Pctsg,11292
-frogml_core/feature_store/data_sources/batch/big_query.py,sha256=h9iscw9I-WyCj1BeyesASNoIiNwPty9acQ37KGD0Ulo,3112
-frogml_core/feature_store/data_sources/batch/clickhouse.py,sha256=mUSE139-ACnInctQNZqETbnWgFyZidvffVJpwQu0T3Q,2149
-frogml_core/feature_store/data_sources/batch/csv.py,sha256=saxFeP--CL4h5j2toyxSuOl4kN_WtEEg8pumaYOjCHg,2070
-frogml_core/feature_store/data_sources/batch/elastic_search.py,sha256=GJKEsOlaxHbBYSor47bofNtV9kJnVVYtq8BSies7hjM,2205
+frogml_core/feature_store/data_sources/batch/_jdbc.py,sha256=u-o7BF2W66ZVLbOhYsL91f1LWThNy8hGog1I-bwyHsU,713
+frogml_core/feature_store/data_sources/batch/athena.py,sha256=VNy6gbIV87vn6TIWDwTBkfcsX0awLCJ5dFZMv448-F0,11410
+frogml_core/feature_store/data_sources/batch/big_query.py,sha256=V5zKuuESbM_0bR_z8FqT-EMWu10GxR3JMdXCZAER8mc,3230
+frogml_core/feature_store/data_sources/batch/clickhouse.py,sha256=HNhTl21BDJ--Bs_5texT_lmTQufBpowJP4ONi-H6Xbw,2267
+frogml_core/feature_store/data_sources/batch/csv.py,sha256=FMeCEVaBWDZKSEmsgF3nk3TRnWp82QzE9DXCYUAEFJs,2187
+frogml_core/feature_store/data_sources/batch/elastic_search.py,sha256=5P9hGg63iV2DWvNWTq-wCXGGJbYzXRs3zU9fPDSfCaY,2323
 frogml_core/feature_store/data_sources/batch/filesystem/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 frogml_core/feature_store/data_sources/batch/filesystem/aws.py,sha256=4mj34C16mH1JeU2dEKFIiV-ubCGHwILlgFNyDuEHHQY,3051
 frogml_core/feature_store/data_sources/batch/filesystem/base_config.py,sha256=fqvqkM_0QY4rcZNa1qB-KLl72o8PhrQQnak8-Bk4tHM,245
 frogml_core/feature_store/data_sources/batch/filesystem/gcp.py,sha256=_UU1a0u1DeTh7okstWiM1e2nD8TJLrEigPIFLQnSkuE,1723
 frogml_core/feature_store/data_sources/batch/filesystem/utils.py,sha256=u2FyAh91JY8j1EzX5fJSgVJqwiwpMvt43mfMeyabVdE,2171
-frogml_core/feature_store/data_sources/batch/mongodb.py,sha256=98Y67LdRgIzfcPNw3CEPV08BOa-HMgrox3Pw-rW883A,1996
-frogml_core/feature_store/data_sources/batch/mysql.py,sha256=MOYgsosXIUf3nap83X4lAn-afFE_tydK5_GU0SW8La8,1736
-frogml_core/feature_store/data_sources/batch/parquet.py,sha256=uOAmTSWn9plN96UuhSqFs_7R9nkrb6DPnJfc7qzcFfY,1963
-frogml_core/feature_store/data_sources/batch/postgres.py,sha256=bpvYg7bgZKG4Mhdh7pp5aN23i9d1GffX5yBSxkQ4Eho,1789
-frogml_core/feature_store/data_sources/batch/redshift.py,sha256=f0xH1Nzg3MMjj-WUFR7DzDLNgawv1KmJQzGSw_-eetE,3268
-frogml_core/feature_store/data_sources/batch/snowflake.py,sha256=T5b9_mRrq_7en-70KgELmgkiKnS3CI9IYw1sBcmSJlw,2671
-frogml_core/feature_store/data_sources/batch/vertica.py,sha256=J9-1KvzWL5JlCqxRYo7MGfF3yXt4wEQ2GzB5RvuS8OM,1905
+frogml_core/feature_store/data_sources/batch/mongodb.py,sha256=sVR38ZBZ0Y8cLng5RVm5ahag0rfABRZ_mngMzJk9YiQ,2111
+frogml_core/feature_store/data_sources/batch/mysql.py,sha256=Hr6urfu6tAa54L-oxTCYugwmuXi2gOtOIIWh-3vBBHU,1813
+frogml_core/feature_store/data_sources/batch/parquet.py,sha256=_KfsDbwJheCtqg1BebmAj6MJTTKlGo9Kd_xpG2upeCk,2081
+frogml_core/feature_store/data_sources/batch/postgres.py,sha256=PoDl3B_gg1wQvLB0NI1vS5hen7rOfd7ShwduWKE93Lk,1870
+frogml_core/feature_store/data_sources/batch/redshift.py,sha256=c5M0L7gmMnmxAvd39D4Eawyx9ZrHfx5YI4p5eMGTbSU,3387
+frogml_core/feature_store/data_sources/batch/snowflake.py,sha256=HWiJoTSe6osG_fITggMjgjttlIOOZxTSRlQ-NGGlbOg,2789
+frogml_core/feature_store/data_sources/batch/vertica.py,sha256=KZo9oIn8vVAYhoE1OWvU_tvcS7vkOr8I2Yj9iDXVd3k,2023
 frogml_core/feature_store/data_sources/source_authentication.py,sha256=AhMxuCYqjmYeaPc5OF_K2YnHq44GcN46AtT3_4nAAIA,962
 frogml_core/feature_store/data_sources/streaming/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 frogml_core/feature_store/data_sources/streaming/_streaming.py,sha256=LM7vcBeuwlpj9JZjAppuy04mtRoaoQtGixv1oaFcJxI,188
 frogml_core/feature_store/data_sources/streaming/kafka/__init__.py,sha256=ISoTTLW8e8hNWJHVekWT0yWhjHEwaNvc3Pyzkd23UAE,670
 frogml_core/feature_store/data_sources/streaming/kafka/authentication.py,sha256=z1VuiHRvALbSFcC2Oa4of2Emy1FHtd6df_-QE6xCR7Q,4305
 frogml_core/feature_store/data_sources/streaming/kafka/deserialization.py,sha256=uKdYTcVOkq8r9rFlimtVqbfCAROQv66Zs1ZQh1yx9jI,3736
-frogml_core/feature_store/data_sources/streaming/kafka/kafka.py,sha256=CcfzXgY6lcxTIEJ7bb3Z8rQM693LtBhd_FSHb-l0xvA,4527
+frogml_core/feature_store/data_sources/streaming/kafka/kafka.py,sha256=UJos6wt_F-DbkvmXAoxZSP8kMlXvF03e9YbyLBR7IkI,4623
 frogml_core/feature_store/data_sources/time_partition_columns.py,sha256=Glt5aMLKY6fCRiNupgxyliKOev_y9fg6c3DYRNDb5hA,6381
 frogml_core/feature_store/entities/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 frogml_core/feature_store/entities/entity.py,sha256=ktMQUgd4Wmfcko9g9MB0KXk7UkLXlJFZpBFD5NRZE_Q,2325
@@ -159,13 +159,13 @@ frogml_core/feature_store/execution/execution_query.py,sha256=xWTVcgfXAxxTHuQyrH
 frogml_core/feature_store/feature_sets/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 frogml_core/feature_store/feature_sets/_utils/_featureset_utils.py,sha256=mnznpQ0RAk5ONRdbp9m4Y95Z2zWUj0bl4ztamtczO1U,1658
 frogml_core/feature_store/feature_sets/backfill.py,sha256=oZqGb8PA1BQxVOxf9-MwzeZyjJqgwulbA-flmBmXAj4,1981
-frogml_core/feature_store/feature_sets/base_feature_set.py,sha256=1niCItXApA6JZB7tviXJJVaw91sFwSBwiA1f2xOQnwM,5378
-frogml_core/feature_store/feature_sets/batch.py,sha256=VADm8KVYhyxV7Ibt2luOlnrbrqd774_ICG0iKPs82cU,17422
+frogml_core/feature_store/feature_sets/base_feature_set.py,sha256=zYJsn0iDBPSS4BmkDyWBKmW-LWJPTWePq9McEu7pNxQ,5415
+frogml_core/feature_store/feature_sets/batch.py,sha256=xnVvONl2sDnOGOiJMYk_uNWACm8oKHl6-CPCZckn-Wc,17636
 frogml_core/feature_store/feature_sets/context.py,sha256=zV6r0O70cfM4pmxlfC6xxAtro-wBhenXWwYwF3KwfTY,263
 frogml_core/feature_store/feature_sets/execution_spec.py,sha256=zKQd7U-PdYkZMqBpA9eIRhhWff-8xxKB_Qo4IDolwGI,2348
 frogml_core/feature_store/feature_sets/metadata.py,sha256=Vv2pyBbwaJZRFhWKRhxdFyN3AsV-DvTQzLs9nyRMWK0,1888
 frogml_core/feature_store/feature_sets/read_policies.py,sha256=BQu6B6IZuKJt8Ff5RYeADdqpHmSkec790RIYeSl6Ulo,6844
-frogml_core/feature_store/feature_sets/streaming.py,sha256=78ItVSojMfLoTY3k0SH6UD40XS7dnVZFn9cKfYUcS3Y,25295
+frogml_core/feature_store/feature_sets/streaming.py,sha256=CDdhFALqWbrU8qAl1iFf0OPF2vWTp53t_hhaWkWMah0,25607
 frogml_core/feature_store/feature_sets/streaming_backfill.py,sha256=u-tjq86AaXAusLUwPYtdCKhTuySQFAtRrUvbPrY3CtI,9834
 frogml_core/feature_store/feature_sets/transformations/__init__.py,sha256=ozc50AI9RBY71nhNiJDu1-vSWJL2Bdgstyh7GGUW2ig,902
 frogml_core/feature_store/feature_sets/transformations/aggregations/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -181,7 +181,7 @@ frogml_core/feature_store/offline/_offline_serving_validations.py,sha256=yfkV8UV
 frogml_core/feature_store/offline/client_v2.py,sha256=kTFyHAYIsKBe3wcuE1S_LyD9CLQ_yCorOLcHVR3Emms,14966
 frogml_core/feature_store/offline/feature_set_features.py,sha256=MjrQrXNhzk7QBdCojdpLfy1fuGdP3GcpOgcc7n7H0G8,740
 frogml_core/feature_store/online/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-frogml_core/feature_store/online/client.py,sha256=_NCVJngFgh8D3IVBW8u8estPqZ7mSwg5IxmuTndFY0A,13572
+frogml_core/feature_store/online/client.py,sha256=U1lpnNE_UGYC2Tvi2y6CayQShYNP0YR8roNI1gxmAWA,13606
 frogml_core/feature_store/online/endpoint_utils.py,sha256=lGssZR-r8kJpcSozVxQAk1_JpVXgRLqOVrK6fw8flPg,2242
 frogml_core/feature_store/sinks/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 frogml_core/feature_store/sinks/base.py,sha256=QdIutDlO_8IBRr-zKfHBRHJ1-DjDmFfR_Yuad193kg0,361
@@ -405,7 +405,7 @@ frogml_core/model/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG
 frogml_core/model/utils/extract_wrapped_function.py,sha256=uIle1zL8vbmeS3PGAuaNFLIUQAsvpuzk3LlH-Teba94,320
 frogml_core/model/utils/feature_utils.py,sha256=ObIU4jLKOMEa3DWEyrXm1m3wC2wseAVwr-zxlSiojYk,2525
 frogml_core/model_loggers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-frogml_core/model_loggers/artifact_logger.py,sha256=7T06QL1keo6BTpIh76JtXRcNlZuIbqECz343kEw0Hfc,3128
+frogml_core/model_loggers/artifact_logger.py,sha256=F3Xhw9O82b0_cx24DHoc13SaJ06rqO5-c4Tw4yIMR74,4033
 frogml_core/model_loggers/data_logger.py,sha256=I3o2TkAGAnKuVWccPMlGL5-rg1e2UDHRc9cRSS8kqmc,5678
 frogml_core/model_loggers/model_logger.py,sha256=xDUIwRzeFAorzzR47ovyag5ohwuYZWp9fRWpLpMzcg4,863
 frogml_core/testing/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -1114,7 +1114,7 @@ frogml_services_mock/mocks/workspace_manager_service_mock.py,sha256=WbOiWgOyr-xT
 frogml_services_mock/services_mock.py,sha256=sgKgwhu2W0YOHtzil8x7f1znK_sZr_i27XSeiF4xqVE,21200
 frogml_services_mock/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 frogml_services_mock/utils/service_utils.py,sha256=ZlB0CnB1J6oBn6_m7fQO2U8tKoboHdUa6ljjkRMYNXU,265
-frogml_storage/__init__.py,sha256=BtvlzlO7UNG5nD9ftqoP0J2wHDhVCXpzuMsjYBw0Wfg,24
+frogml_storage/__init__.py,sha256=oUJHw8__aJg8lCcQGVNPbGho_Gk1e9FvvH2adAponBo,24
 frogml_storage/_environment.py,sha256=zuzOJBtBwFaguwn_JkKjfhXStZoustgP30KzOP3mYv8,707
 frogml_storage/artifactory/__init__.py,sha256=C02rcm7kqsZBVA6c6Gztxamj96hn8Aj6BuzYWFRmWbQ,71
 frogml_storage/artifactory/_artifactory_api.py,sha256=Oz0HOpQPSNwWIVAy94UJUyPhLetc7sdZjoTfSXtrFug,11200
@@ -1153,6 +1153,6 @@ frogml_storage/utils/__init__.py,sha256=HQUWfuGUIPZY7kfS795TRW8BQ4WmNqrNjS7lUrbx
 frogml_storage/utils/_input_checks_utility.py,sha256=CFiJOdTBS9piJMtR3lemEz27wZcQ6_-7XESu8iy-mrw,3221
 frogml_storage/utils/_storage_utils.py,sha256=HB2g7uY5A3b33yIcAUM1OjHb5jWsnpESsiDrEviQwrI,366
 frogml_storage/utils/_url_utils.py,sha256=NUEfz9Fp1iE8b676-A5wrMlSTsJVRKrUhcUItOFAJD8,821
-frogml_core-0.0.111.dist-info/METADATA,sha256=0ex2N3QIInUw4C1Ad9sNbNpL99D99rEMjgp1X84fLPA,14898
-frogml_core-0.0.111.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-frogml_core-0.0.111.dist-info/RECORD,,
+frogml_core-0.0.112.dist-info/METADATA,sha256=q_FfTqj6cc_ch7srz-2ga5qcX-aB2qFJexyE8qtAKts,14898
+frogml_core-0.0.112.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+frogml_core-0.0.112.dist-info/RECORD,,

frogml_storage/__init__.py

@@ -1 +1 @@
-__version__ = "0.0.111"
+__version__ = "0.0.112"