pycontrails 0.60.2__cp312-cp312-manylinux2014_x86_64.manylinux_2_17_x86_64.manylinux_2_28_x86_64.whl

pycontrails/__init__.py

@@ -0,0 +1,70 @@
+"""
+``pycontrails`` public API.
+
+Copyright 2021-present Contrails.org and the Breakthrough Energy Foundation
+
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+   http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
+"""
+
+from __future__ import annotations
+
+import contextlib
+import logging
+from importlib import metadata
+
+import dask.config
+
+# Work around for https://github.com/pydata/xarray/issues/7259
+# Only occurs for xarray 2022.11 and above
+with contextlib.suppress(ImportError):
+    import netCDF4  # noqa: F401
+
+from pycontrails.core.cache import DiskCacheStore, GCPCacheStore
+from pycontrails.core.fleet import Fleet
+from pycontrails.core.flight import Flight, FlightPhase
+from pycontrails.core.fuel import Fuel, HydrogenFuel, JetA, SAFBlend
+from pycontrails.core.met import MetDataArray, MetDataset
+from pycontrails.core.met_var import MetVariable
+from pycontrails.core.models import Model, ModelParams
+from pycontrails.core.vector import GeoVectorDataset, VectorDataset
+from pycontrails.datalib._met_utils.metsource import MetDataSource
+
+__version__ = metadata.version("pycontrails")
+__license__ = "Apache-2.0"
+__url__ = "https://py.contrails.org"
+
+log = logging.getLogger(__name__)
+
+# Hardcode the dask warning silence config
+dask.config.set({"array.slicing.split_large_chunks": False})
+
+
+__all__ = [
+    "DiskCacheStore",
+    "Fleet",
+    "Flight",
+    "FlightPhase",
+    "Fuel",
+    "GCPCacheStore",
+    "GeoVectorDataset",
+    "HydrogenFuel",
+    "JetA",
+    "MetDataArray",
+    "MetDataSource",
+    "MetDataset",
+    "MetVariable",
+    "Model",
+    "ModelParams",
+    "SAFBlend",
+    "VectorDataset",
+]

pycontrails/_version.py

@@ -0,0 +1,34 @@
+# file generated by setuptools-scm
+# don't change, don't track in version control
+
+__all__ = [
+    "__version__",
+    "__version_tuple__",
+    "version",
+    "version_tuple",
+    "__commit_id__",
+    "commit_id",
+]
+
+TYPE_CHECKING = False
+if TYPE_CHECKING:
+    from typing import Tuple
+    from typing import Union
+
+    VERSION_TUPLE = Tuple[Union[int, str], ...]
+    COMMIT_ID = Union[str, None]
+else:
+    VERSION_TUPLE = object
+    COMMIT_ID = object
+
+version: str
+__version__: str
+__version_tuple__: VERSION_TUPLE
+version_tuple: VERSION_TUPLE
+commit_id: COMMIT_ID
+__commit_id__: COMMIT_ID
+
+__version__ = version = '0.60.2'
+__version_tuple__ = version_tuple = (0, 60, 2)
+
+__commit_id__ = commit_id = 'gff1affff2'

pycontrails/core/__init__.py

@@ -0,0 +1,30 @@
+"""Core data structures and methods."""
+
+from pycontrails.core.cache import DiskCacheStore, GCPCacheStore
+from pycontrails.core.fleet import Fleet
+from pycontrails.core.flight import Flight
+from pycontrails.core.fuel import Fuel, HydrogenFuel, JetA, SAFBlend
+from pycontrails.core.met import MetDataArray, MetDataset
+from pycontrails.core.met_var import MetVariable
+from pycontrails.core.models import Model, ModelParams
+from pycontrails.core.vector import GeoVectorDataset, VectorDataset
+from pycontrails.datalib._met_utils.metsource import MetDataSource
+
+__all__ = [
+    "DiskCacheStore",
+    "Fleet",
+    "Flight",
+    "Fuel",
+    "GCPCacheStore",
+    "GeoVectorDataset",
+    "HydrogenFuel",
+    "JetA",
+    "MetDataArray",
+    "MetDataSource",
+    "MetDataset",
+    "MetVariable",
+    "Model",
+    "ModelParams",
+    "SAFBlend",
+    "VectorDataset",
+]