synctatic auto-upgrades (Python 3.9->3.10) TODO: add to .git-blame-ignore-revs

Author: neutrinoceros

yt/_maintenance/backports.py

@@ -63,18 +63,3 @@ def _generate_next_value_(name, start, count, last_values):  # noqa B902
             Return the lower-cased version of the member name.
             """
             return name.lower()
-
-
-builtin_zip = zip
-if sys.version_info >= (3, 10):
-    zip = builtin_zip
-else:
-    # this function is deprecated in more_itertools
-    # because it is superseded by the standard library
-    from more_itertools import zip_equal
-
-    def zip(*args, strict=False):
-        if strict:
-            return zip_equal(*args)
-        else:
-            return builtin_zip(*args)

yt/_maintenance/deprecation.py

@@ -1,15 +1,14 @@
 import warnings
 from functools import wraps
 from types import FunctionType
-from typing import Optional
 
 
 def issue_deprecation_warning(
     msg: str,
     *,
     stacklevel: int,
     since: str,
-    removal: Optional[str] = None,
+    removal: str | None = None,
 ):
     """
     Parameters

yt/_typing.py

@@ -1,34 +1,34 @@
-from typing import Any, Optional, Union
+from typing import Any, Optional
 
 import numpy as np
 import unyt as un
 
-FieldDescT = tuple[str, tuple[str, list[str], Optional[str]]]
+FieldDescT = tuple[str, tuple[str, list[str], str | None]]
 KnownFieldsT = tuple[FieldDescT, ...]
 
 ParticleType = str
 FieldType = str
 FieldName = str
 FieldKey = tuple[FieldType, FieldName]
 ImplicitFieldKey = FieldName
-AnyFieldKey = Union[FieldKey, ImplicitFieldKey]
-DomainDimensions = Union[tuple[int, ...], list[int], np.ndarray]
+AnyFieldKey = FieldKey | ImplicitFieldKey
+DomainDimensions = tuple[int, ...] | list[int] | np.ndarray
 
 ParticleCoordinateTuple = tuple[
     str,  # particle type
     tuple[np.ndarray, np.ndarray, np.ndarray],  # xyz
-    Union[float, np.ndarray],  # hsml
+    float | np.ndarray,  # hsml
 ]
 
 # Geometry specific types
 AxisName = str
 AxisOrder = tuple[AxisName, AxisName, AxisName]
 
 # types that can be converted to un.Unit
-Unit = Union[un.Unit, str]
+Unit = un.Unit | str
 
 # types that can be converted to un.unyt_quantity
-Quantity = Union[un.unyt_quantity, tuple[float, Unit]]
+Quantity = un.unyt_quantity | tuple[float, Unit]
 
 # np.ndarray[...] syntax is runtime-valid from numpy 1.22, we quote it until our minimal
 # runtime requirement is bumped to, or beyond this version

yt/data_objects/analyzer_objects.py

@@ -1,11 +1,7 @@
 import inspect
-import sys
 
 from yt.utilities.object_registries import analysis_task_registry
 
-if sys.version_info < (3, 10):
-    from yt._maintenance.backports import zip
-
 
 class AnalysisTask:
     def __init_subclass__(cls, *args, **kwargs):

yt/data_objects/construction_data_containers.py

@@ -1,7 +1,6 @@
 import fileinput
 import io
 import os
-import sys
 import warnings
 import zipfile
 from functools import partial, wraps
@@ -64,9 +63,6 @@
 )
 from yt.visualization.color_maps import get_colormap_lut
 
-if sys.version_info < (3, 10):
-    from yt._maintenance.backports import zip
-
 
 class YTStreamline(YTSelectionContainer1D):
     """

yt/data_objects/data_containers.py

@@ -1,5 +1,4 @@
 import abc
-import sys
 import weakref
 from collections import defaultdict
 from contextlib import contextmanager
@@ -29,9 +28,6 @@
 from yt.utilities.on_demand_imports import _firefly as firefly
 from yt.utilities.parameter_file_storage import ParameterFileStore
 
-if sys.version_info < (3, 10):
-    from yt._maintenance.backports import zip
-
 if TYPE_CHECKING:
     from yt.data_objects.static_output import Dataset
 

yt/data_objects/derived_quantities.py

@@ -1,5 +1,3 @@
-import sys
-
 import numpy as np
 
 from yt.funcs import camelcase_to_underscore, iter_fields
@@ -13,9 +11,6 @@
 from yt.utilities.physical_constants import gravitational_constant_cgs
 from yt.utilities.physical_ratios import HUGE
 
-if sys.version_info < (3, 10):
-    from yt._maintenance.backports import zip
-
 
 def get_position_fields(field, data):
     axis_names = [data.ds.coordinates.axis_name[num] for num in [0, 1, 2]]

yt/data_objects/level_sets/tests/test_clump_finding.py

@@ -1,6 +1,5 @@
 import os
 import shutil
-import sys
 import tempfile
 
 import numpy as np
@@ -13,9 +12,6 @@
 from yt.testing import requires_file, requires_module
 from yt.utilities.answer_testing.framework import data_dir_load
 
-if sys.version_info < (3, 10):
-    from yt._maintenance.backports import zip
-
 
 def test_clump_finding():
     n_c = 8

yt/data_objects/profiles.py

@@ -1,5 +1,3 @@
-import sys
-
 import numpy as np
 from more_itertools import collapse
 
@@ -25,9 +23,6 @@
     parallel_objects,
 )
 
-if sys.version_info < (3, 10):
-    from yt._maintenance.backports import zip
-
 
 def _sanitize_min_max_units(amin, amax, finfo, registry):
     # returns a copy of amin and amax, converted to finfo's output units

yt/data_objects/static_output.py

@@ -13,7 +13,7 @@
 from functools import cached_property
 from importlib.util import find_spec
 from stat import ST_CTIME
-from typing import Any, Literal, Optional, Union
+from typing import Any, Literal, Optional
 
 import numpy as np
 import unyt as un
@@ -86,15 +86,13 @@
 # to here, and then have it instantiate EnzoDatasets as appropriate.
 
 
-_cached_datasets: MutableMapping[Union[int, str], "Dataset"] = (
-    weakref.WeakValueDictionary()
-)
+_cached_datasets: MutableMapping[int | str, "Dataset"] = weakref.WeakValueDictionary()
 
 # we set this global to None as a place holder
 # its actual instantiation is delayed until after yt.__init__
 # is completed because we need yt.config.ytcfg to be instantiated first
 
-_ds_store: Optional[ParameterFileStore] = None
+_ds_store: ParameterFileStore | None = None
 
 
 def _setup_ds_store(ytcfg: YTConfig) -> None:
@@ -161,24 +159,24 @@ class Dataset(abc.ABC):
     default_field = ("gas", "density")
     fluid_types: tuple[FieldType, ...] = ("gas", "deposit", "index")
     particle_types: tuple[ParticleType, ...] = ("io",)  # By default we have an 'all'
-    particle_types_raw: Optional[tuple[ParticleType, ...]] = ("io",)
+    particle_types_raw: tuple[ParticleType, ...] | None = ("io",)
     geometry: Geometry = Geometry.CARTESIAN
     coordinates = None
     storage_filename = None
-    particle_unions: Optional[dict[ParticleType, ParticleUnion]] = None
-    known_filters: Optional[dict[ParticleType, ParticleFilter]] = None
+    particle_unions: dict[ParticleType, ParticleUnion] | None = None
+    known_filters: dict[ParticleType, ParticleFilter] | None = None
     _index_class: type[Index]
-    field_units: Optional[dict[AnyFieldKey, Unit]] = None
+    field_units: dict[AnyFieldKey, Unit] | None = None
     derived_field_list = requires_index("derived_field_list")
     fields = requires_index("fields")
-    conversion_factors: Optional[dict[str, float]] = None
+    conversion_factors: dict[str, float] | None = None
     # _instantiated represents an instantiation time (since Epoch)
     # the default is a place holder sentinel, falsy value
     _instantiated: float = 0
     _particle_type_counts = None
     _proj_type = "quad_proj"
     _ionization_label_format = "roman_numeral"
-    _determined_fields: Optional[dict[str, list[FieldKey]]] = None
+    _determined_fields: dict[str, list[FieldKey]] | None = None
     fields_detected = False
 
     # these are set in self._parse_parameter_file()
@@ -233,8 +231,8 @@ def __init_subclass__(cls, *args, **kwargs):
     def __init__(
         self,
         filename: str,
-        dataset_type: Optional[str] = None,
-        units_override: Optional[dict[str, str]] = None,
+        dataset_type: str | None = None,
+        units_override: dict[str, str] | None = None,
         # valid unit_system values include all keys from unyt.unit_systems.unit_systems_registry + "code"
         unit_system: Literal[
             "cgs",
@@ -250,7 +248,7 @@ def __init__(
             "Any"
         ] = None,  # Any used as a placeholder here
         *,
-        axis_order: Optional[AxisOrder] = None,
+        axis_order: AxisOrder | None = None,
     ) -> None:
         """
         Base class for generating new output types.  Principally consists of
@@ -732,7 +730,7 @@ def setup_deprecated_fields(self):
             added.append(("gas", old_name))
         self.field_info.find_dependencies(added)
 
-    def _setup_coordinate_handler(self, axis_order: Optional[AxisOrder]) -> None:
+    def _setup_coordinate_handler(self, axis_order: AxisOrder | None) -> None:
         # backward compatibility layer:
         # turning off type-checker on a per-line basis
         cls: type[CoordinateHandler]
@@ -947,7 +945,7 @@ def _setup_particle_types(self, ptypes=None):
 
     def _get_field_info(
         self,
-        field: Union[FieldKey, ImplicitFieldKey, DerivedField],
+        field: FieldKey | ImplicitFieldKey | DerivedField,
         /,
     ) -> DerivedField:
         field_info, candidates = self._get_field_info_helper(field)
@@ -1008,7 +1006,7 @@ def _are_ambiguous(candidates: list[FieldKey]) -> bool:
 
     def _get_field_info_helper(
         self,
-        field: Union[FieldKey, ImplicitFieldKey, DerivedField],
+        field: FieldKey | ImplicitFieldKey | DerivedField,
         /,
     ) -> tuple[DerivedField, list[FieldKey]]:
         self.index
@@ -1280,8 +1278,8 @@ def _assign_unit_system(
         # is mks-like: i.e., it has a current with the same
         # dimensions as amperes.
         mks_system = False
-        mag_unit: Optional[unyt_quantity] = getattr(self, "magnetic_unit", None)
-        mag_dims: Optional[set[Symbol]]
+        mag_unit: unyt_quantity | None = getattr(self, "magnetic_unit", None)
+        mag_dims: set[Symbol] | None
         if mag_unit is not None:
             mag_dims = mag_unit.units.dimensions.free_symbols
         else:
@@ -2057,9 +2055,9 @@ class ParticleFile:
     filename: str
     file_id: int
 
-    start: Optional[int] = None
-    end: Optional[int] = None
-    total_particles: Optional[defaultdict[str, int]] = None
+    start: int | None = None
+    end: int | None = None
+    total_particles: defaultdict[str, int] | None = None
 
     def __init__(self, ds, io, filename, file_id, range=None):
         self.ds = ds