Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
76 changes: 3 additions & 73 deletions src/galax/potential/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -11,44 +11,8 @@
"AbstractCompositePotential",
# composite
"CompositePotential",
# builtin
"LongMuraliBarPotential",
"MonariEtAl2016BarPotential",
"KuzminPotential",
"MiyamotoNagaiPotential",
"MN3ExponentialPotential",
"MN3Sech2Potential",
"HarmonicOscillatorPotential",
"HenonHeilesPotential",
"SatohPotential",
"LMJ09LogarithmicPotential",
"LogarithmicPotential",
"AbstractMultipolePotential",
"MultipoleInnerPotential",
"MultipoleOuterPotential",
"MultipolePotential",
"LeeSutoTriaxialNFWPotential",
"NFWPotential",
"TriaxialNFWPotential",
"Vogelsberger08TriaxialNFWPotential",
"NullPotential",
"BurkertPotential",
"HernquistPotential",
"IsochronePotential",
"JaffePotential",
"KeplerPotential",
"PlummerPotential",
"PowerLawCutoffPotential",
"StoneOstriker15Potential",
"TriaxialHernquistPotential",
"HardCutoffNFWPotential",
"gNFWPotential",
# Pre-composited
"AbstractPreCompositedPotential",
"BovyMWPotential2014",
"LM10Potential",
"MilkyWayPotential",
"MilkyWayPotential2022",
# xfm
"AbstractTransformedPotential",
"FlattenedInThePotential",
Expand Down Expand Up @@ -94,43 +58,8 @@
AbstractPreCompositedPotential,
)
from ._src.base_single import AbstractSinglePotential
from ._src.builtin import (
AbstractMultipolePotential,
BovyMWPotential2014,
BurkertPotential,
HardCutoffNFWPotential,
HarmonicOscillatorPotential,
HenonHeilesPotential,
HernquistPotential,
IsochronePotential,
JaffePotential,
KeplerPotential,
KuzminPotential,
LeeSutoTriaxialNFWPotential,
LM10Potential,
LMJ09LogarithmicPotential,
LogarithmicPotential,
LongMuraliBarPotential,
MilkyWayPotential,
MilkyWayPotential2022,
MiyamotoNagaiPotential,
MN3ExponentialPotential,
MN3Sech2Potential,
MonariEtAl2016BarPotential,
MultipoleInnerPotential,
MultipoleOuterPotential,
MultipolePotential,
NFWPotential,
NullPotential,
PlummerPotential,
PowerLawCutoffPotential,
SatohPotential,
StoneOstriker15Potential,
TriaxialHernquistPotential,
TriaxialNFWPotential,
Vogelsberger08TriaxialNFWPotential,
gNFWPotential,
)
from ._src.builtin import *
from ._src.builtin import __all__ as _builtin_all
Copy link

Copilot AI Aug 13, 2025

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Using wildcard imports (from module import *) is generally discouraged as it makes it unclear what names are being imported and can lead to namespace pollution. Consider using explicit imports or importing the module and accessing attributes through it.

Suggested change
from ._src.builtin import __all__ as _builtin_all
from ._src.builtin import __all__ as _builtin_all
for _name in _builtin_all:
globals()[_name] = getattr(__import__(f"{__name__}._src.builtin", fromlist=[_name]), _name)

Copilot uses AI. Check for mistakes.
from ._src.composite import CompositePotential
from ._src.xfm import (
AbstractTransformedPotential,
Expand All @@ -144,6 +73,7 @@
# isort: split
from ._src import register_funcs

__all__ = __all__ + list(_builtin_all)
Copy link

Copilot AI Aug 13, 2025

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

[nitpick] The expression __all__ = __all__ + list(_builtin_all) modifies __all__ after it's been defined, making the final exported API less obvious. Consider using __all__.extend(_builtin_all) for clearer intent, or define __all__ completely in one place.

Suggested change
__all__ = __all__ + list(_builtin_all)
__all__.extend(_builtin_all)

Copilot uses AI. Check for mistakes.

# Cleanup
del install_import_hook, RUNTIME_TYPECHECKER, register_funcs
44 changes: 22 additions & 22 deletions src/galax/potential/_src/builtin/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -6,41 +6,41 @@
"""

__all__ = [
"LongMuraliBarPotential",
"MonariEtAl2016BarPotential",
"KuzminPotential",
"MiyamotoNagaiPotential",
"MN3ExponentialPotential",
"MN3Sech2Potential",
"AbstractMultipolePotential",
"BovyMWPotential2014",
"BurkertPotential",
"HardCutoffNFWPotential",
"HarmonicOscillatorPotential",
"HenonHeilesPotential",
"SatohPotential",
"HernquistPotential",
"IsochronePotential",
"JaffePotential",
"KeplerPotential",
"KuzminPotential",
"LM10Potential",
"LMJ09LogarithmicPotential",
"LeeSutoTriaxialNFWPotential",
"LogarithmicPotential",
"AbstractMultipolePotential",
"LongMuraliBarPotential",
"MN3ExponentialPotential",
"MN3Sech2Potential",
"MilkyWayPotential",
"MilkyWayPotential2022",
"MiyamotoNagaiPotential",
"MonariEtAl2016BarPotential",
"MultipoleInnerPotential",
"MultipoleOuterPotential",
"MultipolePotential",
"LeeSutoTriaxialNFWPotential",
"NFWPotential",
"TriaxialNFWPotential",
"Vogelsberger08TriaxialNFWPotential",
"gNFWPotential",
"NullPotential",
"BovyMWPotential2014",
"LM10Potential",
"MilkyWayPotential",
"MilkyWayPotential2022",
"BurkertPotential",
"HernquistPotential",
"IsochronePotential",
"JaffePotential",
"KeplerPotential",
"PlummerPotential",
"PowerLawCutoffPotential",
"SatohPotential",
"StoneOstriker15Potential",
"TriaxialHernquistPotential",
"HardCutoffNFWPotential",
"TriaxialNFWPotential",
"Vogelsberger08TriaxialNFWPotential",
"gNFWPotential",
]

from .burkert import BurkertPotential
Expand Down
Loading