shaped 2.0.2__py3-none-any.whl → 2.0.4__py3-none-any.whl

shaped/autogen/models/shaped_internal_recsys_policies_gsasrec_model_policy_gsasrec_model_policy_gsas_rec_model_policy_config.py

@@ -5,7 +5,7 @@
 
     Welcome to Shaped's API reference docs. These provide a  detailed view of the endpoints and CLI commands that Shaped provides and brief  explanations of how they should be used.  The Shaped API has four main endpoints:  **Tables** - Provision and manage batch and real-time data connectors.   **Views** - Configure SQL transformations and AI enrichment on your input data. Use SQL to combine multiple data sources or use an LLM to add new categories, extract specific attributes from descriptions, etc.  **Engines** - Deploy and manage your relevance engines. The Engine API exposes configuration for indexing logic, input datasets, externam embeddings, and more.  **Query** - Execute queries against your engines, to return data based on an input query or rerank an existing list. The Query API exposes the retrieve, filter, score, and ranking steps of the 4-stage ranking architecture.  The base URL for each endpoint is: `https://api.shaped.ai/v2`
 
-    The version of the OpenAPI document: 2.0.0
+    The version of the OpenAPI document: 2.0.4
     Generated by OpenAPI Generator (https://openapi-generator.tech)
 
     Do not edit the class manually.
@@ -58,8 +58,11 @@ class ShapedInternalRecsysPoliciesGsasrecModelPolicyGsasrecModelPolicyGSASRecMod
     eval_after_epochs: Optional[StrictInt] = 1
     split_ratio: Optional[Union[StrictFloat, StrictInt]] = 0.9
     eps: Optional[Union[StrictFloat, StrictInt]] = 1.0E-10
+    normalize_numerical_features: Optional[StrictBool] = Field(default=True, description="Enable NormalizeNumerical transform for entity features.")
+    use_derived_timestamp_features: Optional[StrictBool] = Field(default=True, description="Enable TimestampSinCosEncoder and TimestampCountEncoder for entity features.")
+    balance_labels: Optional[StrictBool] = Field(default=True, description="Enable BalanceLabel transform for interactions.")
     additional_properties: Dict[str, Any] = {}
-    __properties: ClassVar[List[str]] = ["policy_type", "name", "event_values", "batch_size", "eval_batch_size", "n_epochs", "device", "learning_rate", "weight_decay", "patience", "sequence_length", "embedding_dim", "num_heads", "num_blocks", "dropout_rate", "reuse_item_embeddings", "max_batches_per_epoch", "gbce_t", "filter_rated", "neg_per_positive", "eval_after_epochs", "split_ratio", "eps"]
+    __properties: ClassVar[List[str]] = ["policy_type", "name", "event_values", "batch_size", "eval_batch_size", "n_epochs", "device", "learning_rate", "weight_decay", "patience", "sequence_length", "embedding_dim", "num_heads", "num_blocks", "dropout_rate", "reuse_item_embeddings", "max_batches_per_epoch", "gbce_t", "filter_rated", "neg_per_positive", "eval_after_epochs", "split_ratio", "eps", "normalize_numerical_features", "use_derived_timestamp_features", "balance_labels"]
 
     @field_validator('policy_type')
     def policy_type_validate_enum(cls, value):
@@ -193,7 +196,10 @@ class ShapedInternalRecsysPoliciesGsasrecModelPolicyGsasrecModelPolicyGSASRecMod
             "neg_per_positive": NegPerPositive.from_dict(obj["neg_per_positive"]) if obj.get("neg_per_positive") is not None else None,
             "eval_after_epochs": obj.get("eval_after_epochs") if obj.get("eval_after_epochs") is not None else 1,
             "split_ratio": obj.get("split_ratio") if obj.get("split_ratio") is not None else 0.9,
-            "eps": obj.get("eps") if obj.get("eps") is not None else 1.0E-10
+            "eps": obj.get("eps") if obj.get("eps") is not None else 1.0E-10,
+            "normalize_numerical_features": obj.get("normalize_numerical_features") if obj.get("normalize_numerical_features") is not None else True,
+            "use_derived_timestamp_features": obj.get("use_derived_timestamp_features") if obj.get("use_derived_timestamp_features") is not None else True,
+            "balance_labels": obj.get("balance_labels") if obj.get("balance_labels") is not None else True
         })
         # store additional fields in additional_properties
         for _key in obj.keys():

shaped/autogen/models/shaped_internal_recsys_policies_item2vec_model_policy_item2_vec_model_policy_config.py

@@ -5,7 +5,7 @@
 
     Welcome to Shaped's API reference docs. These provide a  detailed view of the endpoints and CLI commands that Shaped provides and brief  explanations of how they should be used.  The Shaped API has four main endpoints:  **Tables** - Provision and manage batch and real-time data connectors.   **Views** - Configure SQL transformations and AI enrichment on your input data. Use SQL to combine multiple data sources or use an LLM to add new categories, extract specific attributes from descriptions, etc.  **Engines** - Deploy and manage your relevance engines. The Engine API exposes configuration for indexing logic, input datasets, externam embeddings, and more.  **Query** - Execute queries against your engines, to return data based on an input query or rerank an existing list. The Query API exposes the retrieve, filter, score, and ranking steps of the 4-stage ranking architecture.  The base URL for each endpoint is: `https://api.shaped.ai/v2`
 
-    The version of the OpenAPI document: 2.0.0
+    The version of the OpenAPI document: 2.0.4
     Generated by OpenAPI Generator (https://openapi-generator.tech)
 
     Do not edit the class manually.

shaped/autogen/models/shaped_internal_recsys_policies_item_content_similarity_model_policy_item_content_similarity_model_policy_config.py

@@ -13,23 +13,27 @@
 
 
 from __future__ import annotations
+
+import json
 import pprint
 import re  # noqa: F401
-import json
+from typing import Any, ClassVar, Dict, List, Optional, Set
 
 from pydantic import BaseModel, ConfigDict, StrictBool, StrictStr, field_validator
-from typing import Any, ClassVar, Dict, List, Optional
 from shaped.autogen.models.distance_function import DistanceFunction
 from shaped.autogen.models.pool_fn import PoolFn
 from shaped.autogen.models.text_encoding import TextEncoding
-from typing import Optional, Set
 from typing_extensions import Self
 
-class ShapedInternalRecsysPoliciesItemContentSimilarityModelPolicyItemContentSimilarityModelPolicyConfig(BaseModel):
+
+class ShapedInternalRecsysPoliciesItemContentSimilarityModelPolicyItemContentSimilarityModelPolicyConfig(
+    BaseModel
+):
     """
     ShapedInternalRecsysPoliciesItemContentSimilarityModelPolicyItemContentSimilarityModelPolicyConfig
-    """ # noqa: E501
-    policy_type: Optional[StrictStr] = 'item-content-similarity'
+    """  # noqa: E501
+
+    policy_type: Optional[StrictStr] = "item-content-similarity"
     name: Optional[StrictStr] = None
     event_values: Optional[List[StrictStr]] = None
     pool_fn: Optional[PoolFn] = None
@@ -40,15 +44,26 @@ class ShapedInternalRecsysPoliciesItemContentSimilarityModelPolicyItemContentSim
     encode_items_using_stored_factors: Optional[StrictBool] = True
     item_feature_names: Optional[List[StrictStr]] = None
     additional_properties: Dict[str, Any] = {}
-    __properties: ClassVar[List[str]] = ["policy_type", "name", "event_values", "pool_fn", "include_item_id", "text_encoding", "categorical_and_numerical_encoding", "distance_fn", "encode_items_using_stored_factors", "item_feature_names"]
-
-    @field_validator('policy_type')
+    __properties: ClassVar[List[str]] = [
+        "policy_type",
+        "name",
+        "event_values",
+        "pool_fn",
+        "include_item_id",
+        "text_encoding",
+        "categorical_and_numerical_encoding",
+        "distance_fn",
+        "encode_items_using_stored_factors",
+        "item_feature_names",
+    ]
+
+    @field_validator("policy_type")
     def policy_type_validate_enum(cls, value):
         """Validates the enum"""
         if value is None:
             return value
 
-        if value not in set(['item-content-similarity']):
+        if value not in set(["item-content-similarity"]):
             raise ValueError("must be one of enum values ('item-content-similarity')")
         return value
 
@@ -58,7 +73,6 @@ class ShapedInternalRecsysPoliciesItemContentSimilarityModelPolicyItemContentSim
         protected_namespaces=(),
     )
 
-
     def to_str(self) -> str:
         """Returns the string representation of the model using alias"""
         return pprint.pformat(self.model_dump(by_alias=True))
@@ -84,9 +98,11 @@ class ShapedInternalRecsysPoliciesItemContentSimilarityModelPolicyItemContentSim
           are ignored.
         * Fields in `self.additional_properties` are added to the output dict.
         """
-        excluded_fields: Set[str] = set([
-            "additional_properties",
-        ])
+        excluded_fields: Set[str] = set(
+            [
+                "additional_properties",
+            ]
+        )
 
         _dict = self.model_dump(
             by_alias=True,
@@ -95,10 +111,10 @@ class ShapedInternalRecsysPoliciesItemContentSimilarityModelPolicyItemContentSim
         )
         # override the default output from pydantic by calling `to_dict()` of pool_fn
         if self.pool_fn:
-            _dict['pool_fn'] = self.pool_fn.to_dict()
+            _dict["pool_fn"] = self.pool_fn.to_dict()
         # override the default output from pydantic by calling `to_dict()` of text_encoding
         if self.text_encoding:
-            _dict['text_encoding'] = self.text_encoding.to_dict()
+            _dict["text_encoding"] = self.text_encoding.to_dict()
         # puts key-value pairs in additional_properties in the top level
         if self.additional_properties is not None:
             for _key, _value in self.additional_properties.items():
@@ -107,17 +123,20 @@ class ShapedInternalRecsysPoliciesItemContentSimilarityModelPolicyItemContentSim
         # set to None if name (nullable) is None
         # and model_fields_set contains the field
         if self.name is None and "name" in self.model_fields_set:
-            _dict['name'] = None
+            _dict["name"] = None
 
         # set to None if event_values (nullable) is None
         # and model_fields_set contains the field
         if self.event_values is None and "event_values" in self.model_fields_set:
-            _dict['event_values'] = None
+            _dict["event_values"] = None
 
         # set to None if item_feature_names (nullable) is None
         # and model_fields_set contains the field
-        if self.item_feature_names is None and "item_feature_names" in self.model_fields_set:
-            _dict['item_feature_names'] = None
+        if (
+            self.item_feature_names is None
+            and "item_feature_names" in self.model_fields_set
+        ):
+            _dict["item_feature_names"] = None
 
         return _dict
 
@@ -130,23 +149,47 @@ class ShapedInternalRecsysPoliciesItemContentSimilarityModelPolicyItemContentSim
         if not isinstance(obj, dict):
             return cls.model_validate(obj)
 
-        _obj = cls.model_validate({
-            "policy_type": obj.get("policy_type") if obj.get("policy_type") is not None else 'item-content-similarity',
-            "name": obj.get("name"),
-            "event_values": obj.get("event_values"),
-            "pool_fn": PoolFn.from_dict(obj["pool_fn"]) if obj.get("pool_fn") is not None else None,
-            "include_item_id": obj.get("include_item_id") if obj.get("include_item_id") is not None else True,
-            "text_encoding": TextEncoding.from_dict(obj["text_encoding"]) if obj.get("text_encoding") is not None else None,
-            "categorical_and_numerical_encoding": obj.get("categorical_and_numerical_encoding") if obj.get("categorical_and_numerical_encoding") is not None else False,
-            "distance_fn": obj.get("distance_fn"),
-            "encode_items_using_stored_factors": obj.get("encode_items_using_stored_factors") if obj.get("encode_items_using_stored_factors") is not None else True,
-            "item_feature_names": obj.get("item_feature_names")
-        })
+        _obj = cls.model_validate(
+            {
+                "policy_type": (
+                    obj.get("policy_type")
+                    if obj.get("policy_type") is not None
+                    else "item-content-similarity"
+                ),
+                "name": obj.get("name"),
+                "event_values": obj.get("event_values"),
+                "pool_fn": (
+                    PoolFn.from_dict(obj["pool_fn"])
+                    if obj.get("pool_fn") is not None
+                    else None
+                ),
+                "include_item_id": (
+                    obj.get("include_item_id")
+                    if obj.get("include_item_id") is not None
+                    else True
+                ),
+                "text_encoding": (
+                    TextEncoding.from_dict(obj["text_encoding"])
+                    if obj.get("text_encoding") is not None
+                    else None
+                ),
+                "categorical_and_numerical_encoding": (
+                    obj.get("categorical_and_numerical_encoding")
+                    if obj.get("categorical_and_numerical_encoding") is not None
+                    else False
+                ),
+                "distance_fn": obj.get("distance_fn"),
+                "encode_items_using_stored_factors": (
+                    obj.get("encode_items_using_stored_factors")
+                    if obj.get("encode_items_using_stored_factors") is not None
+                    else True
+                ),
+                "item_feature_names": obj.get("item_feature_names"),
+            }
+        )
         # store additional fields in additional_properties
         for _key in obj.keys():
             if _key not in cls.__properties:
                 _obj.additional_properties[_key] = obj.get(_key)
 
         return _obj
-
-

shaped/autogen/models/shaped_internal_recsys_policies_lightgbm_model_policy_lightgbm_model_policy_light_gbm_model_policy_config.py

@@ -5,7 +5,7 @@
 
     Welcome to Shaped's API reference docs. These provide a  detailed view of the endpoints and CLI commands that Shaped provides and brief  explanations of how they should be used.  The Shaped API has four main endpoints:  **Tables** - Provision and manage batch and real-time data connectors.   **Views** - Configure SQL transformations and AI enrichment on your input data. Use SQL to combine multiple data sources or use an LLM to add new categories, extract specific attributes from descriptions, etc.  **Engines** - Deploy and manage your relevance engines. The Engine API exposes configuration for indexing logic, input datasets, externam embeddings, and more.  **Query** - Execute queries against your engines, to return data based on an input query or rerank an existing list. The Query API exposes the retrieve, filter, score, and ranking steps of the 4-stage ranking architecture.  The base URL for each endpoint is: `https://api.shaped.ai/v2`
 
-    The version of the OpenAPI document: 2.0.0
+    The version of the OpenAPI document: 2.0.4
     Generated by OpenAPI Generator (https://openapi-generator.tech)
 
     Do not edit the class manually.
@@ -65,8 +65,11 @@ class ShapedInternalRecsysPoliciesLightgbmModelPolicyLightgbmModelPolicyLightGBM
     drop_item_id: Optional[StrictBool] = False
     early_stopping_rounds: Optional[StrictInt] = None
     include_attributes: Optional[List[StrictStr]] = None
+    normalize_numerical_features: Optional[StrictBool] = Field(default=True, description="Enable NormalizeNumerical transform for entity features.")
+    use_derived_timestamp_features: Optional[StrictBool] = Field(default=True, description="Enable TimestampSinCosEncoder and TimestampCountEncoder for entity features.")
+    balance_labels: Optional[StrictBool] = Field(default=True, description="Enable BalanceLabel transform for interactions.")
     additional_properties: Dict[str, Any] = {}
-    __properties: ClassVar[List[str]] = ["policy_type", "name", "event_values", "objective", "n_estimators", "max_depth", "num_leaves", "min_child_weight", "learning_rate", "colsample_bytree", "subsample", "subsample_freq", "zero_as_missing", "bin_construct_sample_cnt", "verbose", "verbose_eval", "num_threads", "enable_resume", "lambdarank_truncation_level", "calibrate", "event_value_user_affinity_features", "event_value_affinity_features_value_filter", "rolling_window_hours", "negative_affinity_features", "content_affinity_features", "content_affinity_features_batch_size", "content_affinity_max_num_latest_items", "content_affinity_embedding_ref", "container_categorical_to_multi_hot", "container_to_container_affinities", "point_in_time_item_feature", "drop_user_id", "drop_item_id", "early_stopping_rounds", "include_attributes"]
+    __properties: ClassVar[List[str]] = ["policy_type", "name", "event_values", "objective", "n_estimators", "max_depth", "num_leaves", "min_child_weight", "learning_rate", "colsample_bytree", "subsample", "subsample_freq", "zero_as_missing", "bin_construct_sample_cnt", "verbose", "verbose_eval", "num_threads", "enable_resume", "lambdarank_truncation_level", "calibrate", "event_value_user_affinity_features", "event_value_affinity_features_value_filter", "rolling_window_hours", "negative_affinity_features", "content_affinity_features", "content_affinity_features_batch_size", "content_affinity_max_num_latest_items", "content_affinity_embedding_ref", "container_categorical_to_multi_hot", "container_to_container_affinities", "point_in_time_item_feature", "drop_user_id", "drop_item_id", "early_stopping_rounds", "include_attributes", "normalize_numerical_features", "use_derived_timestamp_features", "balance_labels"]
 
     @field_validator('policy_type')
     def policy_type_validate_enum(cls, value):
@@ -227,7 +230,10 @@ class ShapedInternalRecsysPoliciesLightgbmModelPolicyLightgbmModelPolicyLightGBM
             "drop_user_id": obj.get("drop_user_id") if obj.get("drop_user_id") is not None else True,
             "drop_item_id": obj.get("drop_item_id") if obj.get("drop_item_id") is not None else False,
             "early_stopping_rounds": obj.get("early_stopping_rounds"),
-            "include_attributes": obj.get("include_attributes")
+            "include_attributes": obj.get("include_attributes"),
+            "normalize_numerical_features": obj.get("normalize_numerical_features") if obj.get("normalize_numerical_features") is not None else True,
+            "use_derived_timestamp_features": obj.get("use_derived_timestamp_features") if obj.get("use_derived_timestamp_features") is not None else True,
+            "balance_labels": obj.get("balance_labels") if obj.get("balance_labels") is not None else True
         })
         # store additional fields in additional_properties
         for _key in obj.keys():

shaped/autogen/models/shaped_internal_recsys_policies_ngram_model_policy_ngram_model_policy_config.py

@@ -5,7 +5,7 @@
 
     Welcome to Shaped's API reference docs. These provide a  detailed view of the endpoints and CLI commands that Shaped provides and brief  explanations of how they should be used.  The Shaped API has four main endpoints:  **Tables** - Provision and manage batch and real-time data connectors.   **Views** - Configure SQL transformations and AI enrichment on your input data. Use SQL to combine multiple data sources or use an LLM to add new categories, extract specific attributes from descriptions, etc.  **Engines** - Deploy and manage your relevance engines. The Engine API exposes configuration for indexing logic, input datasets, externam embeddings, and more.  **Query** - Execute queries against your engines, to return data based on an input query or rerank an existing list. The Query API exposes the retrieve, filter, score, and ranking steps of the 4-stage ranking architecture.  The base URL for each endpoint is: `https://api.shaped.ai/v2`
 
-    The version of the OpenAPI document: 2.0.0
+    The version of the OpenAPI document: 2.0.4
     Generated by OpenAPI Generator (https://openapi-generator.tech)
 
     Do not edit the class manually.

shaped/autogen/models/shaped_internal_recsys_policies_popular_model_policy_popular_model_policy_config.py

@@ -13,35 +13,43 @@
 
 
 from __future__ import annotations
+
+import json
 import pprint
 import re  # noqa: F401
-import json
+from typing import Any, ClassVar, Dict, List, Optional, Set
 
 from pydantic import BaseModel, ConfigDict, StrictStr, field_validator
-from typing import Any, ClassVar, Dict, List, Optional
 from shaped.autogen.models.time_window_in_days import TimeWindowInDays
-from typing import Optional, Set
 from typing_extensions import Self
 
+
 class ShapedInternalRecsysPoliciesPopularModelPolicyPopularModelPolicyConfig(BaseModel):
     """
     ShapedInternalRecsysPoliciesPopularModelPolicyPopularModelPolicyConfig
-    """ # noqa: E501
-    policy_type: Optional[StrictStr] = 'popular'
+    """  # noqa: E501
+
+    policy_type: Optional[StrictStr] = "popular"
     name: Optional[StrictStr] = None
     event_values: Optional[List[StrictStr]] = None
-    mode: Optional[StrictStr] = 'positive_popularity'
+    mode: Optional[StrictStr] = "positive_popularity"
     time_window_in_days: Optional[TimeWindowInDays] = None
     additional_properties: Dict[str, Any] = {}
-    __properties: ClassVar[List[str]] = ["policy_type", "name", "event_values", "mode", "time_window_in_days"]
-
-    @field_validator('policy_type')
+    __properties: ClassVar[List[str]] = [
+        "policy_type",
+        "name",
+        "event_values",
+        "mode",
+        "time_window_in_days",
+    ]
+
+    @field_validator("policy_type")
     def policy_type_validate_enum(cls, value):
         """Validates the enum"""
         if value is None:
             return value
 
-        if value not in set(['popular']):
+        if value not in set(["popular"]):
             raise ValueError("must be one of enum values ('popular')")
         return value
 
@@ -51,7 +59,6 @@ class ShapedInternalRecsysPoliciesPopularModelPolicyPopularModelPolicyConfig(Bas
         protected_namespaces=(),
     )
 
-
     def to_str(self) -> str:
         """Returns the string representation of the model using alias"""
         return pprint.pformat(self.model_dump(by_alias=True))
@@ -77,9 +84,11 @@ class ShapedInternalRecsysPoliciesPopularModelPolicyPopularModelPolicyConfig(Bas
           are ignored.
         * Fields in `self.additional_properties` are added to the output dict.
         """
-        excluded_fields: Set[str] = set([
-            "additional_properties",
-        ])
+        excluded_fields: Set[str] = set(
+            [
+                "additional_properties",
+            ]
+        )
 
         _dict = self.model_dump(
             by_alias=True,
@@ -88,7 +97,7 @@ class ShapedInternalRecsysPoliciesPopularModelPolicyPopularModelPolicyConfig(Bas
         )
         # override the default output from pydantic by calling `to_dict()` of time_window_in_days
         if self.time_window_in_days:
-            _dict['time_window_in_days'] = self.time_window_in_days.to_dict()
+            _dict["time_window_in_days"] = self.time_window_in_days.to_dict()
         # puts key-value pairs in additional_properties in the top level
         if self.additional_properties is not None:
             for _key, _value in self.additional_properties.items():
@@ -97,17 +106,20 @@ class ShapedInternalRecsysPoliciesPopularModelPolicyPopularModelPolicyConfig(Bas
         # set to None if name (nullable) is None
         # and model_fields_set contains the field
         if self.name is None and "name" in self.model_fields_set:
-            _dict['name'] = None
+            _dict["name"] = None
 
         # set to None if event_values (nullable) is None
         # and model_fields_set contains the field
         if self.event_values is None and "event_values" in self.model_fields_set:
-            _dict['event_values'] = None
+            _dict["event_values"] = None
 
         # set to None if time_window_in_days (nullable) is None
         # and model_fields_set contains the field
-        if self.time_window_in_days is None and "time_window_in_days" in self.model_fields_set:
-            _dict['time_window_in_days'] = None
+        if (
+            self.time_window_in_days is None
+            and "time_window_in_days" in self.model_fields_set
+        ):
+            _dict["time_window_in_days"] = None
 
         return _dict
 
@@ -120,18 +132,30 @@ class ShapedInternalRecsysPoliciesPopularModelPolicyPopularModelPolicyConfig(Bas
         if not isinstance(obj, dict):
             return cls.model_validate(obj)
 
-        _obj = cls.model_validate({
-            "policy_type": obj.get("policy_type") if obj.get("policy_type") is not None else 'popular',
-            "name": obj.get("name"),
-            "event_values": obj.get("event_values"),
-            "mode": obj.get("mode") if obj.get("mode") is not None else 'positive_popularity',
-            "time_window_in_days": TimeWindowInDays.from_dict(obj["time_window_in_days"]) if obj.get("time_window_in_days") is not None else None
-        })
+        _obj = cls.model_validate(
+            {
+                "policy_type": (
+                    obj.get("policy_type")
+                    if obj.get("policy_type") is not None
+                    else "popular"
+                ),
+                "name": obj.get("name"),
+                "event_values": obj.get("event_values"),
+                "mode": (
+                    obj.get("mode")
+                    if obj.get("mode") is not None
+                    else "positive_popularity"
+                ),
+                "time_window_in_days": (
+                    TimeWindowInDays.from_dict(obj["time_window_in_days"])
+                    if obj.get("time_window_in_days") is not None
+                    else None
+                ),
+            }
+        )
         # store additional fields in additional_properties
         for _key in obj.keys():
             if _key not in cls.__properties:
                 _obj.additional_properties[_key] = obj.get(_key)
 
         return _obj
-
-

shaped/autogen/models/shaped_internal_recsys_policies_random_model_policy_random_model_policy_config.py

@@ -13,30 +13,32 @@
 
 
 from __future__ import annotations
+
+import json
 import pprint
 import re  # noqa: F401
-import json
+from typing import Any, ClassVar, Dict, List, Optional, Set
 
 from pydantic import BaseModel, ConfigDict, StrictStr, field_validator
-from typing import Any, ClassVar, Dict, List, Optional
-from typing import Optional, Set
 from typing_extensions import Self
 
+
 class ShapedInternalRecsysPoliciesRandomModelPolicyRandomModelPolicyConfig(BaseModel):
     """
     ShapedInternalRecsysPoliciesRandomModelPolicyRandomModelPolicyConfig
-    """ # noqa: E501
-    policy_type: Optional[StrictStr] = 'random'
+    """  # noqa: E501
+
+    policy_type: Optional[StrictStr] = "random"
     name: Optional[StrictStr] = None
     __properties: ClassVar[List[str]] = ["policy_type", "name"]
 
-    @field_validator('policy_type')
+    @field_validator("policy_type")
     def policy_type_validate_enum(cls, value):
         """Validates the enum"""
         if value is None:
             return value
 
-        if value not in set(['random']):
+        if value not in set(["random"]):
             raise ValueError("must be one of enum values ('random')")
         return value
 
@@ -46,7 +48,6 @@ class ShapedInternalRecsysPoliciesRandomModelPolicyRandomModelPolicyConfig(BaseM
         protected_namespaces=(),
     )
 
-
     def to_str(self) -> str:
         """Returns the string representation of the model using alias"""
         return pprint.pformat(self.model_dump(by_alias=True))
@@ -71,8 +72,7 @@ class ShapedInternalRecsysPoliciesRandomModelPolicyRandomModelPolicyConfig(BaseM
           were set at model initialization. Other fields with value `None`
           are ignored.
         """
-        excluded_fields: Set[str] = set([
-        ])
+        excluded_fields: Set[str] = set([])
 
         _dict = self.model_dump(
             by_alias=True,
@@ -82,7 +82,7 @@ class ShapedInternalRecsysPoliciesRandomModelPolicyRandomModelPolicyConfig(BaseM
         # set to None if name (nullable) is None
         # and model_fields_set contains the field
         if self.name is None and "name" in self.model_fields_set:
-            _dict['name'] = None
+            _dict["name"] = None
 
         return _dict
 
@@ -95,10 +95,14 @@ class ShapedInternalRecsysPoliciesRandomModelPolicyRandomModelPolicyConfig(BaseM
         if not isinstance(obj, dict):
             return cls.model_validate(obj)
 
-        _obj = cls.model_validate({
-            "policy_type": obj.get("policy_type") if obj.get("policy_type") is not None else 'random',
-            "name": obj.get("name")
-        })
+        _obj = cls.model_validate(
+            {
+                "policy_type": (
+                    obj.get("policy_type")
+                    if obj.get("policy_type") is not None
+                    else "random"
+                ),
+                "name": obj.get("name"),
+            }
+        )
         return _obj
-
-

shaped/autogen/models/shaped_internal_recsys_policies_recently_popular_policy_recently_popular_policy_config.py

@@ -13,34 +13,38 @@
 
 
 from __future__ import annotations
+
+import json
 import pprint
 import re  # noqa: F401
-import json
+from typing import Any, ClassVar, Dict, List, Optional, Set
 
 from pydantic import BaseModel, ConfigDict, StrictStr, field_validator
-from typing import Any, ClassVar, Dict, List, Optional
 from shaped.autogen.models.mode1 import Mode1
-from typing import Optional, Set
 from typing_extensions import Self
 
-class ShapedInternalRecsysPoliciesRecentlyPopularPolicyRecentlyPopularPolicyConfig(BaseModel):
+
+class ShapedInternalRecsysPoliciesRecentlyPopularPolicyRecentlyPopularPolicyConfig(
+    BaseModel
+):
     """
     ShapedInternalRecsysPoliciesRecentlyPopularPolicyRecentlyPopularPolicyConfig
-    """ # noqa: E501
-    policy_type: Optional[StrictStr] = 'recently-popular'
+    """  # noqa: E501
+
+    policy_type: Optional[StrictStr] = "recently-popular"
     name: Optional[StrictStr] = None
     event_values: Optional[List[StrictStr]] = None
     mode: Optional[Mode1] = None
     additional_properties: Dict[str, Any] = {}
     __properties: ClassVar[List[str]] = ["policy_type", "name", "event_values", "mode"]
 
-    @field_validator('policy_type')
+    @field_validator("policy_type")
     def policy_type_validate_enum(cls, value):
         """Validates the enum"""
         if value is None:
             return value
 
-        if value not in set(['recently-popular']):
+        if value not in set(["recently-popular"]):
             raise ValueError("must be one of enum values ('recently-popular')")
         return value
 
@@ -50,7 +54,6 @@ class ShapedInternalRecsysPoliciesRecentlyPopularPolicyRecentlyPopularPolicyConf
         protected_namespaces=(),
     )
 
-
     def to_str(self) -> str:
         """Returns the string representation of the model using alias"""
         return pprint.pformat(self.model_dump(by_alias=True))
@@ -76,9 +79,11 @@ class ShapedInternalRecsysPoliciesRecentlyPopularPolicyRecentlyPopularPolicyConf
           are ignored.
         * Fields in `self.additional_properties` are added to the output dict.
         """
-        excluded_fields: Set[str] = set([
-            "additional_properties",
-        ])
+        excluded_fields: Set[str] = set(
+            [
+                "additional_properties",
+            ]
+        )
 
         _dict = self.model_dump(
             by_alias=True,
@@ -87,7 +92,7 @@ class ShapedInternalRecsysPoliciesRecentlyPopularPolicyRecentlyPopularPolicyConf
         )
         # override the default output from pydantic by calling `to_dict()` of mode
         if self.mode:
-            _dict['mode'] = self.mode.to_dict()
+            _dict["mode"] = self.mode.to_dict()
         # puts key-value pairs in additional_properties in the top level
         if self.additional_properties is not None:
             for _key, _value in self.additional_properties.items():
@@ -96,12 +101,12 @@ class ShapedInternalRecsysPoliciesRecentlyPopularPolicyRecentlyPopularPolicyConf
         # set to None if name (nullable) is None
         # and model_fields_set contains the field
         if self.name is None and "name" in self.model_fields_set:
-            _dict['name'] = None
+            _dict["name"] = None
 
         # set to None if event_values (nullable) is None
         # and model_fields_set contains the field
         if self.event_values is None and "event_values" in self.model_fields_set:
-            _dict['event_values'] = None
+            _dict["event_values"] = None
 
         return _dict
 
@@ -114,17 +119,25 @@ class ShapedInternalRecsysPoliciesRecentlyPopularPolicyRecentlyPopularPolicyConf
         if not isinstance(obj, dict):
             return cls.model_validate(obj)
 
-        _obj = cls.model_validate({
-            "policy_type": obj.get("policy_type") if obj.get("policy_type") is not None else 'recently-popular',
-            "name": obj.get("name"),
-            "event_values": obj.get("event_values"),
-            "mode": Mode1.from_dict(obj["mode"]) if obj.get("mode") is not None else None
-        })
+        _obj = cls.model_validate(
+            {
+                "policy_type": (
+                    obj.get("policy_type")
+                    if obj.get("policy_type") is not None
+                    else "recently-popular"
+                ),
+                "name": obj.get("name"),
+                "event_values": obj.get("event_values"),
+                "mode": (
+                    Mode1.from_dict(obj["mode"])
+                    if obj.get("mode") is not None
+                    else None
+                ),
+            }
+        )
         # store additional fields in additional_properties
         for _key in obj.keys():
             if _key not in cls.__properties:
                 _obj.additional_properties[_key] = obj.get(_key)
 
         return _obj
-
-