From 6df3071cd9541f0deb3d72db98f9992cc7197498 Mon Sep 17 00:00:00 2001 From: "Jason C. Nucciarone" Date: Mon, 26 Aug 2024 11:37:33 -0400 Subject: [PATCH] chore(lint): use explicit re-exports in module __init__.py Signed-off-by: Jason C. Nucciarone --- slurmutils/editors/__init__.py | 4 ++-- slurmutils/models/__init__.py | 16 +++++++--------- slurmutils/models/slurm.py | 9 +++++++++ 3 files changed, 18 insertions(+), 11 deletions(-) diff --git a/slurmutils/editors/__init__.py b/slurmutils/editors/__init__.py index 3deb773..4bf472e 100644 --- a/slurmutils/editors/__init__.py +++ b/slurmutils/editors/__init__.py @@ -14,5 +14,5 @@ """Editors for Slurm workload manager configuration files.""" -from . import slurmconfig -from . import slurmdbdconfig +from . import slurmconfig as slurmconfig +from . import slurmdbdconfig as slurmdbdconfig diff --git a/slurmutils/models/__init__.py b/slurmutils/models/__init__.py index 65a8922..3ff5c44 100644 --- a/slurmutils/models/__init__.py +++ b/slurmutils/models/__init__.py @@ -14,12 +14,10 @@ """Data models for common Slurm objects.""" -from .slurm import ( - DownNodes, - FrontendNode, - Node, - NodeSet, - Partition, - SlurmConfig, -) -from .slurmdbd import SlurmdbdConfig +from .slurm import DownNodes as DownNodes +from .slurm import FrontendNode as FrontendNode +from .slurm import Node as Node +from .slurm import NodeSet as NodeSet +from .slurm import Partition as Partition +from .slurm import SlurmConfig as SlurmConfig +from .slurmdbd import SlurmdbdConfig as SlurmdbdConfig diff --git a/slurmutils/models/slurm.py b/slurmutils/models/slurm.py index 920c4e3..606345d 100644 --- a/slurmutils/models/slurm.py +++ b/slurmutils/models/slurm.py @@ -14,6 +14,15 @@ """Data models for `slurm.conf` configuration file.""" +__all__ = [ + "Node", + "DownNodes", + "FrontendNode", + "NodeSet", + "Partition", + "SlurmConfig", +] + import copy from typing import Any, Dict, List