pyproject.toml

[build-system]
requires = ["setuptools>=61.0"]
build-backend = "setuptools.build_meta"

[project]
name = "MITIM"
version = "2.1.0"
description = "MIT Integrated Modeling Suite for Fusion Applications"
readme = "README.md"
requires-python = ">=3.9"
license = {file = "LICENSE"}
authors = [
  {name = "P. Rodriguez-Fernandez", email = "pablorf@mit.edu"},
]
classifiers = [
  "Programming Language :: Python :: 3.9",
  "License :: OSI Approved :: MIT License",
  "Topic :: Scientific/Engineering",
  "Operating System :: OS Independent",
]

dependencies = [
  "pip",
  "numpy",
  "matplotlib",
  "argparse",
  "h5py",
  "netCDF4",
  "xarray",
  "pandas",
  "xlsxwriter",
  "statsmodels",
  "dill",
  "IPython",
  "pyDOE",
  "multiprocessing_on_dill",
  "deap",
  "paramiko",
  "tqdm",
  "shapely",
  "freegs",
  "botorch==0.9.4",   # Comes w/ gpytorch==1.11, torch>=1.13.1. PRF also tested w/ torch-2.3.0
  "scikit-image",     # Stricly not for MITIM, but good to have for pygacode
]

[project.optional-dependencies]
pyqt = [
  "PyQt6",
]
omfit = [
  "omfit_classes",
  "scipy<1.14.0",     # As of 08/08/2024, because of https://github.com/gafusion/OMFIT-source/issues/7104
  "numpy<2.0.0",      # For the xarray requirement below to work
  "xarray==2022.6.0", # As of 08/08/2024, because https://github.com/gafusion/OMFIT-source/issues/7104. Froze this version because of the PLASMAstate xr reader (importlib_metadata issues)
  "matplotlib<3.6",   # As of 12/07/2023, omfit_classes fails for higher versions
  "omas",
  "fortranformat",
  "openpyxl",
]
test = [
  "pytest",
  "coverage",
]
nn = [
  "tensorflow==2.15.0", #  As of 08/15/2023, requires python>3.10 (J. Hall)
]

[project.urls]
"Homepage" = "https://mitim-fusion.readthedocs.io/"
"Bug Reports" = "https://github.com/pabloprf/MITIM-fusion/issues"
"Source" = "https://github.com/pabloprf/MITIM-fusion"

[project.scripts]
# mitim_tools interfaces: read, run, plot
mitim_plot_gacode = "mitim_tools.gacode_tools.scripts.read_gacode:main"
mitim_plot_tgyro = "mitim_tools.gacode_tools.scripts.read_tgyro:main"
mitim_plot_tglf = "mitim_tools.gacode_tools.scripts.read_tglf:main"      # [--suffix _0.55] [--gacode input.gacode]
mitim_plot_cgyro = "mitim_tools.gacode_tools.scripts.read_cgyro:main"
mitim_plot_eq = "mitim_tools.gs_tools.scripts.read_eq:main"
mitim_plot_transp = "mitim_tools.transp_tools.scripts.read_transp:main"

mitim_run_tglf = "mitim_tools.gacode_tools.scripts.run_tglf:main"        # (folder input.tglf)  [--gacode input.gacode] [--scan RLTS_2] [--drives True]

# Optimizations
mitim_plot_opt = "mitim_tools.opt_tools.scripts.read:main"               # Not transferred: --type 4 --resolution 20
mitim_plot_portals = "mitim_modules.portals.scripts.read_portals:main"
mitim_slurm = "mitim_tools.opt_tools.scripts.slurm:main"

# TRANSP
mitim_trcheck = "mitim_tools.transp_tools.scripts.run_check:main"	        # e.g. mitim_trcheck pablorf
mitim_trcheck_p = "mitim_tools.transp_tools.scripts.run_check_periodic:main"	# e.g. mitim_trcheck_p pablorf
mitim_trclean = "mitim_tools.transp_tools.scripts.run_clean:main" 		# e.g. mitim_trclean 88664P CMOD --numbers 1,2,3
mitim_trlook = "mitim_tools.transp_tools.scripts.run_look:main"                 # e.g. mitim_trlook 152895P01 CMOD --nofull --plot --remove

# MAESTRO
mitim_plot_maestro = "mitim_modules.maestro.scripts.plot_maestro:main" # --beats 2 (for the last two beats) --only transp

# To run TRANSP (in folder with required files): transp 88664 P01 CMOD --version tshare --trmpi 32 --toricmpi 32 --ptrmpi 32
mitim_run_transp = "mitim_tools.transp_tools.scripts.run_transp:main"

# Others
mitim_compare_nml = "mitim_tools.misc_tools.scripts.compare_namelist:main"
#eff_job="mitim_tools.misc_tools.PARALLELtools.py $1" # Give mitim.out or slurm_output.dat

[tool.pytest.ini_options]
markers = [
]