desc-v2 1.17.4__tar.gz

desc_v2-1.17.4/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2020 Daniel Dudt, Rory Conlin, Dario Panici, Kaya Unalmis, Yigit Gunsur Elmacioglu, Rahul Gaur, Patrick Kim, Egemen Kolemen
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

desc_v2-1.17.4/MANIFEST.in

@@ -0,0 +1,6 @@
+include versioneer.py
+include desc/_version.py
+include desc/examples/*.h5
+
+include LICENSE README.rst pyproject.toml setup.py setup.cfg requirements.txt
+global-exclude __pycache__ *.py[cod] .venv

desc_v2-1.17.4/PKG-INFO

@@ -0,0 +1,190 @@
+Metadata-Version: 2.4
+Name: desc-v2
+Version: 1.17.4
+Summary: Computes, analyzes and optimizes 3D MHD equilibria for stellarators and tokamaks
+Home-page: https://github.com/unalmis/DESC/
+Author: Kaya Unalmis
+Author-email: kunalmis@stanford.edu
+License: MIT
+Project-URL: Issues Tracker, https://github.com/unalmis/DESC/issues
+Project-URL: Contributing, https://github.com/unalmis/DESC/blob/master/CONTRIBUTING.rst
+Project-URL: Source Code, https://github.com/unalmis/DESC
+Project-URL: Documentation, https://unalmis.github.io/DESC/
+Keywords: stellarator tokamak equilibrium perturbation mhd magnetohydrodynamics stability confinement plasma physics optimization design fusion
+Classifier: Development Status :: 3 - Alpha
+Classifier: Intended Audience :: Science/Research
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Natural Language :: English
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.13
+Classifier: Programming Language :: Python :: 3.14
+Classifier: Topic :: Scientific/Engineering
+Classifier: Topic :: Scientific/Engineering :: Mathematics
+Classifier: Topic :: Scientific/Engineering :: Physics
+Classifier: Topic :: Scientific/Engineering :: Visualization
+Requires-Python: >=3.10
+Description-Content-Type: text/x-rst
+License-File: LICENSE
+Requires-Dist: jax!=0.7.1,<0.10.0,>=0.6.2
+Requires-Dist: colorama<=0.4.6
+Requires-Dist: diffrax<=0.7.2,>=0.6.0
+Requires-Dist: equinox<=0.13.8,>=0.11.10
+Requires-Dist: h5py<=3.16.0,>=3.0.0
+Requires-Dist: interpax<0.4,>=0.3.3
+Requires-Dist: interpax_fft<=0.0.9,>=0.0.9
+Requires-Dist: lineax
+Requires-Dist: jax-finufft<=1.3.1,>=1.1.0
+Requires-Dist: matplotlib<=3.10.8,>=3.7.3
+Requires-Dist: mpmath<=1.4.1,>=1.0.0
+Requires-Dist: netcdf4!=1.7.4,<=1.7.5,>=1.5.4
+Requires-Dist: numpy<=2.5,>=1.20.0
+Requires-Dist: nvidia-ml-py>=12.535.77
+Requires-Dist: optax<0.3
+Requires-Dist: orthax<0.3,>=0.2.7
+Requires-Dist: plotly<=6.7.0,>=5.16
+Requires-Dist: psutil<=7.2.2
+Requires-Dist: pylatexenc<=2.10,>=2.0
+Requires-Dist: quadax<0.3,>=0.2.2
+Requires-Dist: scikit-image<=0.26.0
+Requires-Dist: scipy<=1.17.1,>=1.7.0
+Requires-Dist: sympy
+Requires-Dist: termcolor<=3.3.0
+Dynamic: author
+Dynamic: author-email
+Dynamic: classifier
+Dynamic: description
+Dynamic: description-content-type
+Dynamic: home-page
+Dynamic: keywords
+Dynamic: license
+Dynamic: license-file
+Dynamic: project-url
+Dynamic: requires-dist
+Dynamic: requires-python
+Dynamic: summary
+
+.. image:: https://raw.githubusercontent.com/unalmis/DESC/master/docs/_static/images/logo_med_clear.png
+
+.. inclusion-marker-do-not-remove
+
+################################
+Stellarator Optimization Package
+################################
+|License| |DOI| |Issues| |Pypi|
+
+|Docs| |UnitTests| |RegressionTests|
+
+Improvements
+============
+Like the original package, but with bounce integrals as well as all the improvements
+in `this fork <https://github.com/unalmis/DESC>`__.
+
+A list of papers which feature DESC can be found here_.
+
+.. _Theory: https://unalmis.github.io/DESC/theory_general.html
+.. _here: https://unalmis.github.io/DESC/pubs_list.html
+
+
+Quick Start
+===========
+
+The easiest way to install DESC is from PyPI: ``pip install desc-v2``
+
+For more detailed instructions on installing DESC and its dependencies, see Installation_.
+
+The best place to start learning about DESC is our tutorials:
+
+- `Basic fixed boundary equilibrium`_: running from a VMEC input, creating an equilibrium from scratch
+- `Advanced equilibrium`_: continuation and perturbation methods.
+- `Free boundary equilibrium`_: vacuum and or finite beta with external field.
+- `Using DESC outputs`_: analysis, plotting, saving to VMEC format.
+- `Basic optimization`_: specifying objectives, fixing degrees of freedom.
+- `Advanced optimization`_: advanced constraints, precise quasi-symmetry, constrained optimization.
+- `Near axis constraints`_: loading solutions from QSC/QIC and fixing near axis expansion.
+- `Coil optimization`_: "second stage" optimization of magnetic coils.
+
+For details on the various objectives, constraints, optimizable objects and more, see
+the full `api documentation`_.
+
+If all you need is an equilibrium solution, the simplest method is through the command
+line by giving an input file specifying the equilibrium and solver options, this
+way can also can also accept VMEC input files.
+
+The code is run using the syntax ``desc <path/to/inputfile>`` and the full list
+of command line options are given in `Command Line Interface`_. (Note that you may have
+to prepend the command with ``python -m``)
+
+Refer to `Inputs`_ for documentation on how to format the input file.
+
+The equilibrium solution is output in a HDF5 binary file, whose format is detailed in `Outputs`_.
+
+.. _Installation: https://unalmis.github.io/DESC/installation.html
+.. _Command Line Interface: https://unalmis.github.io/DESC/command_line.html
+.. _Inputs: https://unalmis.github.io/DESC/input.html
+.. _Outputs: https://unalmis.github.io/DESC/output.html
+.. _Basic fixed boundary equilibrium: https://unalmis.github.io/DESC/notebooks/tutorials/basic_equilibrium.html
+.. _Advanced equilibrium: https://unalmis.github.io/DESC/notebooks/tutorials/advanced_equilibrium_continuation.html
+.. _Free boundary equilibrium: https://unalmis.github.io/DESC/notebooks/tutorials/free_boundary_equilibrium.html
+.. _Using DESC outputs: https://unalmis.github.io/DESC/notebooks/tutorials/use_outputs.html
+.. _Basic optimization: https://unalmis.github.io/DESC/notebooks/tutorials/basic_optimization.html
+.. _Advanced optimization: https://unalmis.github.io/DESC/notebooks/tutorials/advanced_optimization.html
+.. _Near axis constraints: https://unalmis.github.io/DESC/notebooks/tutorials/nae_constraint.html
+.. _Coil optimization: https://unalmis.github.io/DESC/notebooks/tutorials/coil_stage_two_optimization.html
+.. _api documentation: https://unalmis.github.io/DESC/api.html
+
+Repository Contents
+===================
+
+- desc_ contains the source code including the main script and supplemental files. Refer to the API_ documentation for details on all of the available functions and classes.
+- docs_ contains the documentation files.
+- tests_ contains routines for automatic testing.
+- publications_ contains PDFs of publications by the DESC group, as well as scripts and data to reproduce the results of these papers.
+
+.. _desc: https://github.com/unalmis/DESC/tree/master/desc
+.. _docs: https://github.com/unalmis/DESC/tree/master/docs
+.. _tests: https://github.com/unalmis/DESC/tree/master/tests
+.. _publications: https://github.com/unalmis/DESC/tree/master/publications
+.. _API: https://unalmis.github.io/DESC/api.html
+
+Contribute
+==========
+
+- `Contributing guidelines <https://github.com/unalmis/DESC/blob/master/CONTRIBUTING.rst>`_
+- `Issue Tracker <https://github.com/unalmis/DESC/issues>`_
+- `Source Code <https://github.com/unalmis/DESC/>`_
+- `Documentation <https://unalmis.github.io/DESC/>`_
+
+.. |License| image:: https://img.shields.io/github/license/unalmis/DESC?color=blue&logo=open-source-initiative&logoColor=white
+    :target: https://github.com/unalmis/DESC/blob/master/LICENSE
+    :alt: License
+
+.. |DOI| image:: https://zenodo.org/badge/DOI/10.5281/zenodo.4876504.svg
+   :target: https://doi.org/10.5281/zenodo.4876504
+   :alt: DOI
+
+.. |Docs| image:: https://github.com/unalmis/DESC/actions/workflows/docs.yml/badge.svg
+    :target: https://unalmis.github.io/DESC/
+    :alt: Documentation
+
+.. |UnitTests| image:: https://github.com/unalmis/DESC/actions/workflows/unit_tests.yml/badge.svg
+    :target: https://github.com/unalmis/DESC/actions/workflows/unit_tests.yml
+    :alt: UnitTests
+
+.. |RegressionTests| image:: https://github.com/unalmis/DESC/actions/workflows/regression_tests.yml/badge.svg
+    :target: https://github.com/unalmis/DESC/actions/workflows/regression_tests.yml
+    :alt: RegressionTests
+
+.. |Codecov| image:: https://codecov.io/gh/unalmis/DESC/branch/master/graph/badge.svg
+    :target: https://codecov.io/gh/unalmis/DESC
+    :alt: Coverage
+
+.. |Issues| image:: https://img.shields.io/github/issues/unalmis/DESC
+    :target: https://github.com/unalmis/DESC/issues
+    :alt: GitHub issues
+
+.. |Pypi| image:: https://img.shields.io/pypi/v/desc-v2
+    :target: https://pypi.org/project/desc-v2/
+    :alt: Pypi

desc_v2-1.17.4/README.rst

@@ -0,0 +1,122 @@
+.. image:: https://raw.githubusercontent.com/unalmis/DESC/master/docs/_static/images/logo_med_clear.png
+
+.. inclusion-marker-do-not-remove
+
+################################
+Stellarator Optimization Package
+################################
+|License| |DOI| |Issues| |Pypi|
+
+|Docs| |UnitTests| |RegressionTests|
+
+Improvements
+============
+Like the original package, but with bounce integrals as well as all the improvements
+in `this fork <https://github.com/unalmis/DESC>`__.
+
+A list of papers which feature DESC can be found here_.
+
+.. _Theory: https://unalmis.github.io/DESC/theory_general.html
+.. _here: https://unalmis.github.io/DESC/pubs_list.html
+
+
+Quick Start
+===========
+
+The easiest way to install DESC is from PyPI: ``pip install desc-v2``
+
+For more detailed instructions on installing DESC and its dependencies, see Installation_.
+
+The best place to start learning about DESC is our tutorials:
+
+- `Basic fixed boundary equilibrium`_: running from a VMEC input, creating an equilibrium from scratch
+- `Advanced equilibrium`_: continuation and perturbation methods.
+- `Free boundary equilibrium`_: vacuum and or finite beta with external field.
+- `Using DESC outputs`_: analysis, plotting, saving to VMEC format.
+- `Basic optimization`_: specifying objectives, fixing degrees of freedom.
+- `Advanced optimization`_: advanced constraints, precise quasi-symmetry, constrained optimization.
+- `Near axis constraints`_: loading solutions from QSC/QIC and fixing near axis expansion.
+- `Coil optimization`_: "second stage" optimization of magnetic coils.
+
+For details on the various objectives, constraints, optimizable objects and more, see
+the full `api documentation`_.
+
+If all you need is an equilibrium solution, the simplest method is through the command
+line by giving an input file specifying the equilibrium and solver options, this
+way can also can also accept VMEC input files.
+
+The code is run using the syntax ``desc <path/to/inputfile>`` and the full list
+of command line options are given in `Command Line Interface`_. (Note that you may have
+to prepend the command with ``python -m``)
+
+Refer to `Inputs`_ for documentation on how to format the input file.
+
+The equilibrium solution is output in a HDF5 binary file, whose format is detailed in `Outputs`_.
+
+.. _Installation: https://unalmis.github.io/DESC/installation.html
+.. _Command Line Interface: https://unalmis.github.io/DESC/command_line.html
+.. _Inputs: https://unalmis.github.io/DESC/input.html
+.. _Outputs: https://unalmis.github.io/DESC/output.html
+.. _Basic fixed boundary equilibrium: https://unalmis.github.io/DESC/notebooks/tutorials/basic_equilibrium.html
+.. _Advanced equilibrium: https://unalmis.github.io/DESC/notebooks/tutorials/advanced_equilibrium_continuation.html
+.. _Free boundary equilibrium: https://unalmis.github.io/DESC/notebooks/tutorials/free_boundary_equilibrium.html
+.. _Using DESC outputs: https://unalmis.github.io/DESC/notebooks/tutorials/use_outputs.html
+.. _Basic optimization: https://unalmis.github.io/DESC/notebooks/tutorials/basic_optimization.html
+.. _Advanced optimization: https://unalmis.github.io/DESC/notebooks/tutorials/advanced_optimization.html
+.. _Near axis constraints: https://unalmis.github.io/DESC/notebooks/tutorials/nae_constraint.html
+.. _Coil optimization: https://unalmis.github.io/DESC/notebooks/tutorials/coil_stage_two_optimization.html
+.. _api documentation: https://unalmis.github.io/DESC/api.html
+
+Repository Contents
+===================
+
+- desc_ contains the source code including the main script and supplemental files. Refer to the API_ documentation for details on all of the available functions and classes.
+- docs_ contains the documentation files.
+- tests_ contains routines for automatic testing.
+- publications_ contains PDFs of publications by the DESC group, as well as scripts and data to reproduce the results of these papers.
+
+.. _desc: https://github.com/unalmis/DESC/tree/master/desc
+.. _docs: https://github.com/unalmis/DESC/tree/master/docs
+.. _tests: https://github.com/unalmis/DESC/tree/master/tests
+.. _publications: https://github.com/unalmis/DESC/tree/master/publications
+.. _API: https://unalmis.github.io/DESC/api.html
+
+Contribute
+==========
+
+- `Contributing guidelines <https://github.com/unalmis/DESC/blob/master/CONTRIBUTING.rst>`_
+- `Issue Tracker <https://github.com/unalmis/DESC/issues>`_
+- `Source Code <https://github.com/unalmis/DESC/>`_
+- `Documentation <https://unalmis.github.io/DESC/>`_
+
+.. |License| image:: https://img.shields.io/github/license/unalmis/DESC?color=blue&logo=open-source-initiative&logoColor=white
+    :target: https://github.com/unalmis/DESC/blob/master/LICENSE
+    :alt: License
+
+.. |DOI| image:: https://zenodo.org/badge/DOI/10.5281/zenodo.4876504.svg
+   :target: https://doi.org/10.5281/zenodo.4876504
+   :alt: DOI
+
+.. |Docs| image:: https://github.com/unalmis/DESC/actions/workflows/docs.yml/badge.svg
+    :target: https://unalmis.github.io/DESC/
+    :alt: Documentation
+
+.. |UnitTests| image:: https://github.com/unalmis/DESC/actions/workflows/unit_tests.yml/badge.svg
+    :target: https://github.com/unalmis/DESC/actions/workflows/unit_tests.yml
+    :alt: UnitTests
+
+.. |RegressionTests| image:: https://github.com/unalmis/DESC/actions/workflows/regression_tests.yml/badge.svg
+    :target: https://github.com/unalmis/DESC/actions/workflows/regression_tests.yml
+    :alt: RegressionTests
+
+.. |Codecov| image:: https://codecov.io/gh/unalmis/DESC/branch/master/graph/badge.svg
+    :target: https://codecov.io/gh/unalmis/DESC
+    :alt: Coverage
+
+.. |Issues| image:: https://img.shields.io/github/issues/unalmis/DESC
+    :target: https://github.com/unalmis/DESC/issues
+    :alt: GitHub issues
+
+.. |Pypi| image:: https://img.shields.io/pypi/v/desc-v2
+    :target: https://pypi.org/project/desc-v2/
+    :alt: Pypi

desc_v2-1.17.4/desc/__init__.py

@@ -0,0 +1,191 @@
+"""DESC: a 3D MHD equilibrium solver and stellarator optimization suite."""
+
+import importlib
+import os
+import re
+import warnings
+
+import colorama
+from termcolor import colored
+
+from ._version import get_versions
+
+__version__ = get_versions()["version"]
+del get_versions
+
+colorama.init()
+
+
+__all__ = [
+    "basis",
+    "coils",
+    "compute",
+    "continuation",
+    "derivatives",
+    "equilibrium",
+    "examples",
+    "geometry",
+    "grid",
+    "io",
+    "magnetic_fields",
+    "objectives",
+    "optimize",
+    "particles",
+    "perturbations",
+    "plotting",
+    "profiles",
+    "random",
+    "transform",
+    "vmec",
+]
+
+
+def __getattr__(name):
+    if name in __all__:
+        return importlib.import_module("." + name, __name__)
+    raise AttributeError(f"module {__name__!r} has no attribute {name!r}")
+
+
+_BANNER = r"""
+ ____  ____  _____   ___
+|  _ \| ___|/  ___|/ ___|
+| | \ | |_  | (__ | |
+| | | |  _| \___ \| |
+| |_/ | |__  ___) | |___
+|____/|____||____/ \____|
+
+"""
+
+BANNER = colored(_BANNER, "magenta")
+
+
+config = {"device": None, "avail_mem": None, "kind": None}
+
+
+def set_device(kind="cpu", gpuid=None):
+    """Sets the device to use for computation.
+
+    If kind==``'gpu'`` and a gpuid is specified, uses the specified GPU. If
+    gpuid==``None`` or a wrong GPU id is given, checks available GPUs and selects the
+    one with the most available memory.
+    Respects environment variable CUDA_VISIBLE_DEVICES for selecting from multiple
+    available GPUs
+
+    Parameters
+    ----------
+    kind : {``'cpu'``, ``'gpu'``}
+        whether to use CPU or GPU.
+
+    """
+    config["kind"] = kind
+    if kind == "cpu":
+        os.environ["JAX_PLATFORMS"] = "cpu"
+        os.environ["CUDA_VISIBLE_DEVICES"] = ""
+        import psutil
+
+        cpu_mem = psutil.virtual_memory().available / 1024**3  # RAM in GB
+        config["device"] = "CPU"
+        config["avail_mem"] = cpu_mem
+
+    if kind == "gpu":
+        # Set CUDA_DEVICE_ORDER so the IDs assigned by CUDA match those from nvidia-smi
+        os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"
+        # pynvml namespace is exposed through nvidia-ml-py
+        from pynvml import (
+            nvmlDeviceGetCount,
+            nvmlDeviceGetHandleByIndex,
+            nvmlDeviceGetMemoryInfo,
+            nvmlDeviceGetName,
+            nvmlDeviceGetUUID,
+            nvmlInit,
+            nvmlMemory_v2,
+            nvmlShutdown,
+        )
+
+        def _gpu_info():
+            """Equivalent to nvgpu.gpu_info() using nvidia-ml-py."""
+            nvmlInit()
+            try:
+                info = []
+                for device_idx in range(nvmlDeviceGetCount()):
+                    handle = nvmlDeviceGetHandleByIndex(device_idx)
+                    # Use nvmlMemory_v2 to account for system-reserved memory
+                    mem = nvmlDeviceGetMemoryInfo(handle, version=nvmlMemory_v2)
+                    _bytes_to_mib = 1024 * 1024
+                    mem_used = mem.used // _bytes_to_mib
+                    mem_total = mem.total // _bytes_to_mib
+                    info.append(
+                        {
+                            "index": str(device_idx),
+                            "type": nvmlDeviceGetName(handle),
+                            "uuid": nvmlDeviceGetUUID(handle),
+                            "mem_used": mem_used,
+                            "mem_total": mem_total,
+                            "mem_used_percent": 100.0 * mem_used / mem_total,
+                        }
+                    )
+                return info
+            finally:
+                nvmlShutdown()
+
+        try:
+            devices = _gpu_info()
+        except FileNotFoundError:
+            devices = []
+        if len(devices) == 0:
+            warnings.warn(colored("No GPU found, falling back to CPU", "yellow"))
+            set_device(kind="cpu")
+            return
+
+        maxmem = 0
+        selected_gpu = None
+        gpu_ids = [dev["index"] for dev in devices]
+        if "CUDA_VISIBLE_DEVICES" in os.environ:
+            cuda_ids = [
+                s for s in re.findall(r"\b\d+\b", os.environ["CUDA_VISIBLE_DEVICES"])
+            ]
+            # check that the visible devices actually exist and are gpus
+            gpu_ids = [i for i in cuda_ids if i in gpu_ids]
+        if len(gpu_ids) == 0:
+            # cuda visible devices = '' -> don't use any gpu
+            warnings.warn(
+                colored(
+                    (
+                        "CUDA_VISIBLE_DEVICES={} ".format(
+                            os.environ["CUDA_VISIBLE_DEVICES"]
+                        )
+                        + "did not match any physical GPU "
+                        + "(id={}), falling back to CPU".format(
+                            [dev["index"] for dev in devices]
+                        )
+                    ),
+                    "yellow",
+                )
+            )
+            set_device(kind="cpu")
+            return
+        devices = [dev for dev in devices if dev["index"] in gpu_ids]
+
+        if gpuid is not None and (str(gpuid) in gpu_ids):
+            selected_gpu = [dev for dev in devices if dev["index"] == str(gpuid)][0]
+        else:
+            for dev in devices:
+                mem = dev["mem_total"] - dev["mem_used"]
+                if mem > maxmem:
+                    maxmem = mem
+                    selected_gpu = dev
+        config["device"] = selected_gpu["type"] + " (id={})".format(
+            selected_gpu["index"]
+        )
+        if gpuid is not None and not (str(gpuid) in gpu_ids):
+            warnings.warn(
+                colored(
+                    "Specified gpuid {} not found, falling back to ".format(str(gpuid))
+                    + config["device"],
+                    "yellow",
+                )
+            )
+        config["avail_mem"] = (
+            selected_gpu["mem_total"] - selected_gpu["mem_used"]
+        ) / 1024  # in GB
+        os.environ["CUDA_VISIBLE_DEVICES"] = str(selected_gpu["index"])

desc_v2-1.17.4/desc/__main__.py

@@ -0,0 +1,92 @@
+"""Main command line interface to DESC for solving fixed boundary equilibria."""
+
+import sys
+
+from desc.input_reader import InputReader
+
+
+def main(cl_args=sys.argv[1:]):
+    """Run the main DESC code from the command line.
+
+    Reads and parses user input from command line, runs the code,
+    and prints and plots the resulting equilibrium.
+
+    """
+    ir = InputReader(cl_args=cl_args)
+
+    if ir.args.version:
+        return
+
+    import desc
+
+    if ir.args.verbose:
+        print(desc.BANNER)
+
+    import matplotlib.pyplot as plt
+
+    from desc.backend import print_backend_info
+    from desc.equilibrium import EquilibriaFamily, Equilibrium
+    from desc.plotting import plot_section, plot_surfaces
+
+    if ir.args.verbose:
+        print_backend_info()
+        print("Reading input from {}".format(ir.input_path))
+        print("Outputs will be written to {}".format(ir.output_path))
+
+    inputs = ir.inputs
+    if (
+        len(inputs) == 1
+        and (inputs[-1]["pres_ratio"] is None)
+        and (inputs[-1]["bdry_ratio"] is None)
+    ):
+        eq = Equilibrium(**inputs[-1], check_kwargs=False, ensure_nested=False)
+        equil_fam = EquilibriaFamily.solve_continuation_automatic(
+            eq,
+            objective=inputs[-1]["objective"],
+            optimizer=inputs[-1]["optimizer"],
+            pert_order=inputs[-1]["pert_order"],
+            ftol=inputs[-1]["ftol"],
+            xtol=inputs[-1]["xtol"],
+            gtol=inputs[-1]["gtol"],
+            maxiter=inputs[-1]["maxiter"],
+            verbose=ir.args.verbose,
+            checkpoint_path=ir.output_path,
+        )
+    else:
+        # initialize
+        equil_fam = EquilibriaFamily(inputs)
+        # check vmec path input
+        if ir.args.guess is not None:
+            if ir.args.verbose:
+                print("Initial guess from {}".format(ir.args.guess))
+            equil_fam[0].set_initial_guess(ir.args.guess)
+        # solve equilibrium
+        equil_fam.solve_continuation(
+            objective=inputs[0]["objective"],
+            optimizer=inputs[0]["optimizer"],
+            pert_order=[inp["pert_order"] for inp in inputs],
+            ftol=[inp["ftol"] for inp in inputs],
+            xtol=[inp["xtol"] for inp in inputs],
+            gtol=[inp["gtol"] for inp in inputs],
+            maxiter=[inp["maxiter"] for inp in inputs],
+            verbose=ir.args.verbose,
+            checkpoint_path=ir.output_path,
+        )
+
+    if ir.args.plot > 1:
+        for i, eq in enumerate(equil_fam[:-1]):
+            print("Plotting solution at step {}".format(i + 1))
+            _ = plot_surfaces(eq)
+            plt.show()
+            _ = plot_section(eq, "|F|_normalized", log=True)
+            plt.show()
+    if ir.args.plot > 0:
+        print("Plotting final solution")
+        _ = plot_surfaces(equil_fam[-1])
+        plt.show()
+        _ = plot_section(equil_fam[-1], "|F|_normalized", log=True)
+        plt.show()
+
+
+if __name__ == "__main__":  # pragma: no cover
+    main(sys.argv[1:])

desc_v2-1.17.4/desc/_version.py

@@ -0,0 +1,21 @@
+
+# This file was generated by 'versioneer.py' (0.19) from
+# revision-control system data, or from the parent directory name of an
+# unpacked source archive. Distribution tarballs contain a pre-generated copy
+# of this file.
+
+import json
+
+version_json = '''
+{
+ "date": "2026-06-21T16:37:47-0700",
+ "dirty": false,
+ "error": null,
+ "full-revisionid": "49cf9484f963b2bdbddd63e3cd7a5bf5735394ee",
+ "version": "1.17.4"
+}
+'''  # END VERSION_JSON
+
+
+def get_versions():
+    return json.loads(version_json)