fixed flake8 lint errors due to version upgrade

.pre-commit-config.yaml

@@ -13,7 +13,7 @@ repos:
     hooks:
       - id: black
         additional_dependencies: ['click==8.0.4']
-        language_version: python3.9
+        language_version: python3.11
   - repo: https://github.com/PyCQA/flake8
     rev: '6.0.0'
     hooks:

src/zipline/currency.py

@@ -61,7 +61,7 @@ def name(self):
         return self._name
 
     def __eq__(self, other):
-        if type(self) != type(other):
+        if type(self) is not type(other):
             return NotImplemented
         return self.code == other.code
 

src/zipline/finance/trading.py

@@ -35,9 +35,8 @@ def __init__(
         data_frequency="daily",
         arena="backtest",
     ):
-
-        assert type(start_session) == pd.Timestamp
-        assert type(end_session) == pd.Timestamp
+        assert type(start_session) is pd.Timestamp
+        assert type(end_session) is pd.Timestamp
 
         assert trading_calendar is not None, "Must pass in trading calendar!"
         assert start_session <= end_session, "Period start falls after period end."

src/zipline/testing/core.py

@@ -293,7 +293,7 @@ def check_allclose(actual, desired, rtol=1e-07, atol=0, err_msg="", verbose=True
     --------
     np.assert_allclose
     """
-    if type(actual) != type(desired):
+    if type(actual) is not type(desired):
         raise AssertionError("%s != %s" % (type(actual), type(desired)))
     return assert_allclose(
         actual,
@@ -313,7 +313,7 @@ def check_arrays(x, y, err_msg="", verbose=True, check_dtypes=True):
     --------
     np.assert_array_equal
     """
-    assert type(x) == type(y), "{x} != {y}".format(x=type(x), y=type(y))
+    assert type(x) is type(y), "{x} != {y}".format(x=type(x), y=type(y))
     assert x.dtype == y.dtype, "{x.dtype} != {y.dtype}".format(x=x, y=y)
 
     if isinstance(x, LabelArray):

src/zipline/testing/fixtures.py

@@ -1017,7 +1017,6 @@ class level fixtures.
 
     BCOLZ_DAILY_BAR_PATH = "daily_equity_pricing.bcolz"
     BCOLZ_DAILY_BAR_READ_ALL_THRESHOLD = None
-    BCOLZ_DAILY_BAR_COUNTRY_CODE = None
     EQUITY_DAILY_BAR_SOURCE_FROM_MINUTE = False
     # allows WithBcolzEquityDailyBarReaderFromCSVs to call the
     # `write_csvs`method without needing to reimplement `init_class_fixtures`

src/zipline/testing/predicates.py

@@ -544,7 +544,7 @@ def assert_timestamp_and_datetime_equal(
 
     Returns raises unless ``allow_datetime_coercions`` is passed as True.
     """
-    assert allow_datetime_coercions or type(result) == type(expected), (
+    assert allow_datetime_coercions or type(result) is type(expected), (
         "%sdatetime types (%s, %s) don't match and "
         "allow_datetime_coercions was not set.\n%s"
         % (

src/zipline/utils/factory.py

@@ -42,7 +42,7 @@ def create_simulation_parameters(
 
     if start is None:
         start = pd.Timestamp(f"{year}-01-01", tz="UTC")
-    elif type(start) == datetime:
+    elif type(start) is datetime:
         start = pd.Timestamp(start)
 
     if end is None:
@@ -51,7 +51,7 @@ def create_simulation_parameters(
             end = trading_calendar.sessions[start_index + num_days - 1]
         else:
             end = pd.Timestamp(f"{year}-12-31", tz="UTC")
-    elif type(end) == datetime:
+    elif type(end) is datetime:
         end = pd.Timestamp(end)
 
     sim_params = SimulationParameters(