Merge pull request #2107 from cta-observatory/py311

Support python 3.11

.github/workflows/ci.yml

@@ -46,6 +46,10 @@ jobs:
             install-method: mamba
             extra-args: ["codecov"]
 
+          - os: ubuntu-latest
+            python-version: "3.11"
+            install-method: pip
+
           - os: ubuntu-latest
             python-version: "3.10"
             install-method: pip

ctapipe/compat.py

@@ -0,0 +1,17 @@
+"""
+Module for python version compatibility
+"""
+import sys
+
+__all__ = [
+    "StrEnum",
+]
+
+
+if sys.version_info >= (3, 11):
+    from enum import StrEnum
+else:
+    from enum import Enum
+
+    class StrEnum(str, Enum):
+        """Compatibility backfill of StrEnum for python < 3.11"""

ctapipe/instrument/optics.py

@@ -9,6 +9,7 @@
 import numpy as np
 from astropy.table import QTable
 
+from ..compat import StrEnum
 from ..utils import get_table_dataset
 
 logger = logging.getLogger(__name__)
@@ -37,7 +38,7 @@ class FocalLengthKind(Enum):
 
 
 @unique
-class SizeType(str, Enum):
+class SizeType(StrEnum):
     """
     Enumeration of different telescope sizes (LST, MST, SST)
     """

ctapipe/reco/reconstructor.py

@@ -1,6 +1,5 @@
 import weakref
 from abc import abstractmethod
-from enum import Enum
 
 import astropy.units as u
 import joblib
@@ -11,6 +10,7 @@
 from ctapipe.core import Provenance, QualityQuery, TelescopeComponent
 from ctapipe.core.traits import List
 
+from ..compat import StrEnum
 from ..coordinates import shower_impact_distance
 
 __all__ = [
@@ -22,7 +22,7 @@
 ]
 
 
-class ReconstructionProperty(str, Enum):
+class ReconstructionProperty(StrEnum):
     """
     Primary particle properties estimated by a `Reconstructor`
 

ctapipe/tests/test_compat.py

@@ -0,0 +1,9 @@
+from ctapipe.compat import StrEnum
+
+
+def test_str_enum():
+    class Foo(StrEnum):
+        A = "A"
+        B = "B"
+
+    assert f"{Foo.A}" == "A"

docs/changes/2107.maintenance.rst

@@ -0,0 +1 @@
+Add support for python 3.11.

docs/conf.py

@@ -119,6 +119,7 @@ def setup(app):
     ("py:class", "astropy.coordinates.baseframe.BaseCoordinateFrame"),
     ("py:class", "astropy.table.table.Table"),
     ("py:class", "eventio.simtel.simtelfile.SimTelFile"),
+    ("py:class", "ctapipe.compat.StrEnum"),
 ]
 
 # The suffix(es) of source filenames.

setup.cfg

@@ -33,7 +33,7 @@ install_requires=
     importlib_metadata  ; python_version < "3.10"
     joblib
     matplotlib ~=3.0
-    numba ~=0.56.0
+    numba >=0.56
     numpy ~=1.16
     psutil
     pyyaml >=5.1