lint

Author: mrlevitas

reddit_decider/__init__.py

@@ -5,6 +5,11 @@
 from dataclasses import dataclass
 from datetime import timedelta
 from enum import Enum
+from rust_decider import Decider as RustDecider
+from rust_decider import DeciderException
+from rust_decider import Decision
+from rust_decider import FeatureNotFoundException
+from rust_decider import ValueTypeMismatchException
 from typing import Any
 from typing import Callable
 from typing import cast
@@ -25,11 +30,6 @@
 from baseplate.lib.file_watcher import FileWatcher
 from baseplate.lib.file_watcher import WatchedFileNotAvailableError
 from reddit_edgecontext import ValidatedAuthenticationToken
-from rust_decider import Decider as RustDecider
-from rust_decider import DeciderException
-from rust_decider import Decision
-from rust_decider import FeatureNotFoundException
-from rust_decider import ValueTypeMismatchException
 from typing_extensions import Literal
 
 from .prometheus_metrics import experiments_client_counter

tests/decider_tests.py

@@ -1521,19 +1521,14 @@ def test_range_variant_emit_event_override(self):
                 "type": "feature_rollout",
                 "emit_event": False,
                 "experiment": {
-                "variants": [
-                    {
-                    "name": "enabled",
-                    "size": 1.0,
-                    "range_end": 1.0,
-                    "range_start": 0.0
-                    }
-                ],
-                "experiment_version": 1,
-                "shuffle_version": 0,
-                "bucket_val": "user_id",
-                "log_bucketing": False
-                }
+                    "variants": [
+                        {"name": "enabled", "size": 1.0, "range_end": 1.0, "range_start": 0.0}
+                    ],
+                    "experiment_version": 1,
+                    "shuffle_version": 0,
+                    "bucket_val": "user_id",
+                    "log_bucketing": False,
+                },
             },
             "measured_rollout_100": {
                 "id": 9119,
@@ -1547,21 +1542,21 @@ def test_range_variant_emit_event_override(self):
                 "emit_event": False,
                 "measured": True,
                 "experiment": {
-                "variants": [
-                    {
-                    "name": "enabled",
-                    "size": 1.0,
-                    "range_end": 1.0,
-                    "range_start": 0.0,
-                    "emit_event_override": True
-                    }
-                ],
-                "experiment_version": 1,
-                "shuffle_version": 0,
-                "bucket_val": "user_id",
-                "log_bucketing": False
-                }
-            }
+                    "variants": [
+                        {
+                            "name": "enabled",
+                            "size": 1.0,
+                            "range_end": 1.0,
+                            "range_start": 0.0,
+                            "emit_event_override": True,
+                        }
+                    ],
+                    "experiment_version": 1,
+                    "shuffle_version": 0,
+                    "bucket_val": "user_id",
+                    "log_bucketing": False,
+                },
+            },
         }
 
         with create_temp_config_file(cfg) as f:
@@ -1573,7 +1568,7 @@ def test_range_variant_emit_event_override(self):
 
             # FR does NOT emit exposure
             self.assertEqual(self.event_logger.log.call_count, 0)
-            
+
             # FR is NOT "measured"
             fr_cfg = decider.get_experiment(experiment_name="feature_rollout_100")
             self.assertEqual(fr_cfg.measured, False)
@@ -1583,7 +1578,7 @@ def test_range_variant_emit_event_override(self):
 
             # Measured Rollout DOES emit exposure (due to RV "emit_event_override" field)
             self.assertEqual(self.event_logger.log.call_count, 1)
-            
+
             # MR IS "measured"
             mr_cfg = decider.get_experiment(experiment_name="measured_rollout_100")
             self.assertEqual(mr_cfg.measured, True)