Fixing formatting issues

Author: keelerm84

ldclient/impl/integrations/test_datav2/test_data_sourcev2.py

@@ -1,5 +1,4 @@
 import threading
-import time
 from queue import Empty, Queue
 from typing import Generator
 

ldclient/integrations/test_datav2.py

@@ -37,8 +37,8 @@ class FlagRuleBuilderV2:
 
     def __init__(self, flag_builder: FlagBuilderV2):
         self._flag_builder = flag_builder
-        self._clauses = []  # type: List[dict]
-        self._variation = None  # type: Optional[int]
+        self._clauses: List[dict] = []
+        self._variation: Optional[int] = None
 
     def and_match(self, attribute: str, *values) -> FlagRuleBuilderV2:
         """
@@ -170,11 +170,11 @@ def __init__(self, key: str):
         """:param str key: The name of the flag"""
         self._key = key
         self._on = True
-        self._variations = []  # type: List[Any]
-        self._off_variation = None  # type: Optional[int]
-        self._fallthrough_variation = None  # type: Optional[int]
-        self._targets = {}  # type: Dict[str, Dict[int, Set[str]]]
-        self._rules = []  # type: List[FlagRuleBuilderV2]
+        self._variations: List[Any] = []
+        self._off_variation: Optional[int] = None
+        self._fallthrough_variation: Optional[int] = None
+        self._targets: Dict[str, Dict[int, Set[str]]] = {}
+        self._rules: List[FlagRuleBuilderV2] = []
 
     # Note that _copy is private by convention, because we don't want developers to
     # consider it part of the public API, but it is still called from TestDataV2.
@@ -703,4 +703,4 @@ def build_synchronizer(self) -> _TestDataSourceV2:
 
         :return: a test data synchronizer
         """
-        return _TestDataSourceV2(self)
+        return _TestDataSourceV2(self)

ldclient/testing/integrations/test_test_data_sourcev2.py

@@ -4,7 +4,6 @@
 
 import pytest
 
-from ldclient.context import Context
 from ldclient.impl.datasystem.protocolv2 import (
     ChangeType,
     IntentCode,
@@ -399,10 +398,10 @@ def test_initializer_can_sync():
     """Test that an initializer can call sync() and get initial data"""
     td = TestDataV2.data_source()
     td.update(td.flag('test-flag').variation_for_all(True))
-    
+
     initializer = td.build_initializer()
     sync_gen = initializer.sync()
-    
+
     # Should get initial update with data
     initial_update = next(sync_gen)
     assert initial_update.state == DataSourceState.VALID