[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

Author: pre-commit-ci[bot]

unstract/task-abstraction/src/unstract/task_abstraction/backends/hatchet.py

@@ -278,7 +278,9 @@ def create_steps(self):
 
                         # Capture loop variables in closure by using default arguments
                         def create_workflow_step(step_obj, parent_list):
-                            @self.hatchet.step(name=step_obj.task_name, parents=parent_list)
+                            @self.hatchet.step(
+                                name=step_obj.task_name, parents=parent_list
+                            )
                             def workflow_step(context):
                                 # Get the original task function
                                 task_fn = self._tasks[step_obj.task_name]
@@ -289,10 +291,13 @@ def workflow_step(context):
                                     workflow_input = context.step_output(parent_list[0])
                                 else:
                                     # Use initial workflow input
-                                    workflow_input = context.workflow_input()["initial_input"]
+                                    workflow_input = context.workflow_input()[
+                                        "initial_input"
+                                    ]
 
                                 # Execute task with input and step kwargs
                                 return task_fn(workflow_input, **step_obj.kwargs)
+
                             return workflow_step
 
                         workflow_step = create_workflow_step(step, parents)

unstract/task-abstraction/tests/integration/test_backend_selection.py

@@ -134,6 +134,7 @@ async def test_backend_selection_by_feature_flags(
                 def create_mock_flag_response(tc):
                     def mock_flag_response(flag_key, namespace, entity_id, context=None):
                         return tc.feature_flags.get(flag_key, False)
+
                     return mock_flag_response
 
                 mock_flag.side_effect = create_mock_flag_response(test_case)
@@ -169,14 +170,19 @@ async def test_rollout_percentage_distribution(self, backend_selector):
             ) as mock_flag:
                 # Mock percentage-based rollout - capture scenario in closure
                 def create_mock_percentage_rollout(scen):
-                    def mock_percentage_rollout(flag_key, namespace, entity_id, context=None):
+                    def mock_percentage_rollout(
+                        flag_key, namespace, entity_id, context=None
+                    ):
                         if flag_key == "task_abstraction_enabled":
                             import hashlib
 
-                            hash_value = int(hashlib.md5(entity_id.encode()).hexdigest(), 16)
+                            hash_value = int(
+                                hashlib.md5(entity_id.encode()).hexdigest(), 16
+                            )
                             user_bucket = hash_value % 100
                             return user_bucket < scen["percentage"]
                         return False
+
                     return mock_percentage_rollout
 
                 mock_flag.side_effect = create_mock_percentage_rollout(scenario)
@@ -226,21 +232,24 @@ async def test_organization_based_selection(self, backend_selector):
             with patch(
                 "unstract.flags.feature_flag.check_feature_flag_status"
             ) as mock_flag:
-
                 # Capture case in closure
                 def create_mock_org_based_flags(c):
-                    def mock_org_based_flags(flag_key, namespace, entity_id, context=None):
+                    def mock_org_based_flags(
+                        flag_key, namespace, entity_id, context=None
+                    ):
                         org_id = context.get("organization_id") if context else None
 
                         # Organization-specific logic
                         if org_id == "beta_org" and flag_key == "hatchet_backend_enabled":
                             return True
                         elif (
-                            org_id == "stable_org" and flag_key == "task_abstraction_enabled"
+                            org_id == "stable_org"
+                            and flag_key == "task_abstraction_enabled"
                         ):
                             return True
 
                         return c["feature_flags"].get(flag_key, False)
+
                     return mock_org_based_flags
 
                 mock_flag.side_effect = create_mock_org_based_flags(case)
@@ -269,11 +278,11 @@ async def test_fallback_chain_construction(
             with patch(
                 "unstract.flags.feature_flag.check_feature_flag_status"
             ) as mock_flag:
-
                 # Capture test_case in closure
                 def create_mock_flag_response(tc):
                     def mock_flag_response(flag_key, namespace, entity_id, context=None):
                         return tc.feature_flags.get(flag_key, False)
+
                     return mock_flag_response
 
                 mock_flag.side_effect = create_mock_flag_response(test_case)
@@ -344,7 +353,6 @@ async def test_user_segment_based_selection(self, backend_selector):
             with patch(
                 "unstract.flags.feature_flag.check_feature_flag_status"
             ) as mock_flag:
-
                 # Capture segment in closure
                 def create_mock_segment_based_flags(seg):
                     def mock_segment_based_flags(
@@ -362,6 +370,7 @@ def mock_segment_based_flags(
                             elif seg["segment"] == "free_users":
                                 return flag_key == "unified_celery_enabled"
                         return False
+
                     return mock_segment_based_flags
 
                 mock_flag.side_effect = create_mock_segment_based_flags(segment)
@@ -405,11 +414,11 @@ async def test_workflow_specific_backend_preferences(self, backend_selector):
             with patch(
                 "unstract.flags.feature_flag.check_feature_flag_status"
             ) as mock_flag:
-
                 # Capture preference in closure
                 def create_mock_flag_response(pref):
                     def mock_flag_response(flag_key, namespace, entity_id, context=None):
                         return pref["feature_flags"].get(flag_key, False)
+
                     return mock_flag_response
 
                 mock_flag.side_effect = create_mock_flag_response(preference)

unstract/task-abstraction/tests/integration/test_feature_flag_rollout.py

@@ -139,12 +139,17 @@ async def test_percentage_based_rollout(
             ) as mock_flag:
                 # Mock percentage-based rollout - capture percentage in closure
                 def create_mock_percentage_rollout(pct):
-                    def mock_percentage_rollout(flag_key, namespace, entity_id, context=None):
+                    def mock_percentage_rollout(
+                        flag_key, namespace, entity_id, context=None
+                    ):
                         if flag_key == "task_abstraction_enabled":
-                            hash_value = int(hashlib.md5(entity_id.encode()).hexdigest(), 16)
+                            hash_value = int(
+                                hashlib.md5(entity_id.encode()).hexdigest(), 16
+                            )
                             user_bucket = hash_value % 100
                             return user_bucket < pct
                         return False
+
                     return mock_percentage_rollout
 
                 mock_flag.side_effect = create_mock_percentage_rollout(percentage)
@@ -332,7 +337,9 @@ def mock_progressive_rollout(
                     first_enabled_index = i
                 elif first_enabled_index is not None and not enabled:
                     # User was disabled after being enabled - this shouldn't happen
-                    pytest.fail(f"User {user_id} was disabled after being enabled at stage {rollout_stages[first_enabled_index]}%")
+                    pytest.fail(
+                        f"User {user_id} was disabled after being enabled at stage {rollout_stages[first_enabled_index]}%"
+                    )
 
     @pytest.mark.asyncio
     async def test_rollback_scenario(self, feature_flag_manager):