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

shaped/autogen/models/score_ensemble_policy_config.py

@@ -13,33 +13,40 @@
 
 
 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 ScoreEnsemblePolicyConfig(BaseModel):
     """
     Configuration for ScoreEnsemblePolicy.  This policy combines multiple scoring policies using a value model to determine the final scores.
-    """ # noqa: E501
-    policy_type: Optional[StrictStr] = 'score-ensemble'
+    """  # noqa: E501
+
+    policy_type: Optional[StrictStr] = "score-ensemble"
     name: Optional[StrictStr] = None
     policies: Optional[List[ScoreEnsemblePolicyConfigPoliciesInner]] = None
     value_model: Optional[StrictStr] = None
     additional_properties: Dict[str, Any] = {}
-    __properties: ClassVar[List[str]] = ["policy_type", "name", "policies", "value_model"]
-
-    @field_validator('policy_type')
+    __properties: ClassVar[List[str]] = [
+        "policy_type",
+        "name",
+        "policies",
+        "value_model",
+    ]
+
+    @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(['score-ensemble']):
+        if value not in set(["score-ensemble"]):
             raise ValueError("must be one of enum values ('score-ensemble')")
         return value
 
@@ -49,7 +56,6 @@ class ScoreEnsemblePolicyConfig(BaseModel):
         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))
@@ -75,9 +81,11 @@ class ScoreEnsemblePolicyConfig(BaseModel):
           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,
@@ -90,7 +98,7 @@ class ScoreEnsemblePolicyConfig(BaseModel):
             for _item_policies in self.policies:
                 if _item_policies:
                     _items.append(_item_policies.to_dict())
-            _dict['policies'] = _items
+            _dict["policies"] = _items
         # 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():
@@ -99,17 +107,17 @@ class ScoreEnsemblePolicyConfig(BaseModel):
         # 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 policies (nullable) is None
         # and model_fields_set contains the field
         if self.policies is None and "policies" in self.model_fields_set:
-            _dict['policies'] = None
+            _dict["policies"] = None
 
         # set to None if value_model (nullable) is None
         # and model_fields_set contains the field
         if self.value_model is None and "value_model" in self.model_fields_set:
-            _dict['value_model'] = None
+            _dict["value_model"] = None
 
         return _dict
 
@@ -122,12 +130,25 @@ class ScoreEnsemblePolicyConfig(BaseModel):
         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 'score-ensemble',
-            "name": obj.get("name"),
-            "policies": [ScoreEnsemblePolicyConfigPoliciesInner.from_dict(_item) for _item in obj["policies"]] if obj.get("policies") is not None else None,
-            "value_model": obj.get("value_model")
-        })
+        _obj = cls.model_validate(
+            {
+                "policy_type": (
+                    obj.get("policy_type")
+                    if obj.get("policy_type") is not None
+                    else "score-ensemble"
+                ),
+                "name": obj.get("name"),
+                "policies": (
+                    [
+                        ScoreEnsemblePolicyConfigPoliciesInner.from_dict(_item)
+                        for _item in obj["policies"]
+                    ]
+                    if obj.get("policies") is not None
+                    else None
+                ),
+                "value_model": obj.get("value_model"),
+            }
+        )
         # store additional fields in additional_properties
         for _key in obj.keys():
             if _key not in cls.__properties:
@@ -135,7 +156,10 @@ class ScoreEnsemblePolicyConfig(BaseModel):
 
         return _obj
 
-from shaped.autogen.models.score_ensemble_policy_config_policies_inner import ScoreEnsemblePolicyConfigPoliciesInner
+
+from shaped.autogen.models.score_ensemble_policy_config_policies_inner import (
+    ScoreEnsemblePolicyConfigPoliciesInner,
+)
+
 # TODO: Rewrite to not use raise_errors
 ScoreEnsemblePolicyConfig.model_rebuild(raise_errors=False)
-