REFACTOR-#6889: Define __all__ in modin.config.__init__.py (#6886)

Signed-off-by: Anatoly Myachev <anatoly.myachev@intel.com>

modin/config/__init__.py

@@ -13,5 +13,95 @@
 
 """Module houses config entities which can be used for Modin behavior tuning."""
 
-from .envvars import *  # noqa: F403, F401
-from .pubsub import Parameter  # noqa: F401
+from modin.config.envvars import (
+    AsvDataSizeConfig,
+    AsvImplementation,
+    AsyncReadMode,
+    BenchmarkMode,
+    CIAWSAccessKeyID,
+    CIAWSSecretAccessKey,
+    CpuCount,
+    DoUseCalcite,
+    Engine,
+    EnvironmentVariable,
+    ExperimentalGroupbyImpl,
+    ExperimentalNumPyAPI,
+    GithubCI,
+    GpuCount,
+    HdkFragmentSize,
+    HdkLaunchParameters,
+    IsDebug,
+    IsExperimental,
+    IsRayCluster,
+    LogFileSize,
+    LogMemoryInterval,
+    LogMode,
+    Memory,
+    MinPartitionSize,
+    ModinNumpy,
+    NPartitions,
+    PersistentPickle,
+    ProgressBar,
+    RangePartitioningGroupby,
+    RayRedisAddress,
+    RayRedisPassword,
+    ReadSqlEngine,
+    StorageFormat,
+    TestDatasetSize,
+    TestRayClient,
+    TestReadFromPostgres,
+    TestReadFromSqlServer,
+    TrackFileLeaks,
+)
+from modin.config.pubsub import Parameter, ValueSource
+
+__all__ = [
+    "EnvironmentVariable",
+    "Parameter",
+    "ValueSource",
+    # General settings
+    "IsDebug",
+    "Engine",
+    "StorageFormat",
+    "CpuCount",
+    "GpuCount",
+    "Memory",
+    # Ray specific
+    "IsRayCluster",
+    "RayRedisAddress",
+    "RayRedisPassword",
+    "TestRayClient",
+    # Partitioning
+    "NPartitions",
+    "MinPartitionSize",
+    # HDK specific
+    "HdkFragmentSize",
+    "DoUseCalcite",
+    "HdkLaunchParameters",
+    # ASV specific
+    "TestDatasetSize",
+    "AsvImplementation",
+    "AsvDataSizeConfig",
+    # Specific features
+    "ProgressBar",
+    "BenchmarkMode",
+    "PersistentPickle",
+    "ModinNumpy",
+    "ExperimentalNumPyAPI",
+    "RangePartitioningGroupby",
+    "ExperimentalGroupbyImpl",
+    "AsyncReadMode",
+    "ReadSqlEngine",
+    "IsExperimental",
+    # For tests
+    "TrackFileLeaks",
+    "TestReadFromSqlServer",
+    "TestReadFromPostgres",
+    "GithubCI",
+    "CIAWSSecretAccessKey",
+    "CIAWSAccessKeyID",
+    # Logging
+    "LogMode",
+    "LogMemoryInterval",
+    "LogFileSize",
+]

modin/config/__main__.py

@@ -24,15 +24,18 @@
 
 import pandas
 
-from . import *  # noqa: F403, F401
-from .pubsub import Parameter
+import modin.config as cfg
 
 
 def print_config_help() -> None:
     """Print configs help messages."""
-    for objname in sorted(globals()):
-        obj = globals()[objname]
-        if isinstance(obj, type) and issubclass(obj, Parameter) and not obj.is_abstract:
+    for objname in sorted(cfg.__all__):
+        obj = getattr(cfg, objname)
+        if (
+            isinstance(obj, type)
+            and issubclass(obj, cfg.Parameter)
+            and not obj.is_abstract
+        ):
             print(f"{obj.get_help()}\n\tCurrent value: {obj.get()}")  # noqa: T201
 
 
@@ -51,9 +54,13 @@ def export_config_help(filename: str) -> None:
         CpuCount="multiprocessing.cpu_count()",
         NPartitions="equals to MODIN_CPUS env",
     )
-    for objname in sorted(globals()):
-        obj = globals()[objname]
-        if isinstance(obj, type) and issubclass(obj, Parameter) and not obj.is_abstract:
+    for objname in sorted(cfg.__all__):
+        obj = getattr(cfg, objname)
+        if (
+            isinstance(obj, type)
+            and issubclass(obj, cfg.Parameter)
+            and not obj.is_abstract
+        ):
             data = {
                 "Config Name": obj.__name__,
                 "Env. Variable Name": getattr(

modin/config/envvars.py

@@ -23,7 +23,7 @@
 from packaging import version
 from pandas.util._decorators import doc  # type: ignore[attr-defined]
 
-from .pubsub import (
+from modin.config.pubsub import (
     _TYPE_PARAMS,
     _UNSET,
     DeprecationDescriptor,

modin/config/pubsub.py

@@ -28,7 +28,7 @@
 )
 
 if TYPE_CHECKING:
-    from .envvars import EnvironmentVariable
+    from modin.config.envvars import EnvironmentVariable
 
 
 class DeprecationDescriptor:

modin/config/test/test_envvars.py

@@ -19,16 +19,11 @@
 from packaging import version
 
 import modin.config as cfg
-from modin.config.envvars import (
-    _UNSET,
-    EnvironmentVariable,
-    ExactStr,
-    Parameter,
-    _check_vars,
-)
+from modin.config.envvars import _check_vars
+from modin.config.pubsub import _UNSET, ExactStr
 
 
-def reset_vars(*vars: tuple[Parameter]):
+def reset_vars(*vars: tuple[cfg.Parameter]):
     """
     Reset value for the passed parameters.
 
@@ -51,7 +46,7 @@ def make_unknown_env():
 
 @pytest.fixture(params=[str, ExactStr])
 def make_custom_envvar(request):
-    class CustomVar(EnvironmentVariable, type=request.param):
+    class CustomVar(cfg.EnvironmentVariable, type=request.param):
         """custom var"""
 
         default = 10

modin/config/test/test_parameter.py

@@ -15,7 +15,7 @@
 
 import pytest
 
-from modin.config.pubsub import Parameter
+from modin.config import Parameter
 
 
 def make_prefilled(vartype, varinit):

modin/pandas/__init__.py

@@ -101,9 +101,13 @@
 
 
 def _update_engine(publisher: Parameter):
-    from modin.config import CpuCount, Engine, StorageFormat
-    from modin.config.envvars import IsExperimental
-    from modin.config.pubsub import ValueSource
+    from modin.config import (
+        CpuCount,
+        Engine,
+        IsExperimental,
+        StorageFormat,
+        ValueSource,
+    )
 
     # Set this so that Pandas doesn't try to multithread by itself
     os.environ["OMP_NUM_THREADS"] = "1"

modin/pandas/test/test_groupby.py

@@ -21,8 +21,12 @@
 import pytest
 
 import modin.pandas as pd
-from modin.config import NPartitions, StorageFormat
-from modin.config.envvars import IsRayCluster, RangePartitioningGroupby
+from modin.config import (
+    IsRayCluster,
+    NPartitions,
+    RangePartitioningGroupby,
+    StorageFormat,
+)
 from modin.core.dataframe.algebra.default2pandas.groupby import GroupBy
 from modin.core.dataframe.pandas.partitioning.axis_partition import (
     PandasDataframeAxisPartition,

modin/pandas/test/test_io.py

@@ -39,13 +39,13 @@
     AsyncReadMode,
     Engine,
     IsExperimental,
+    MinPartitionSize,
     ReadSqlEngine,
     StorageFormat,
     TestDatasetSize,
     TestReadFromPostgres,
     TestReadFromSqlServer,
 )
-from modin.config.envvars import MinPartitionSize
 from modin.db_conn import ModinDatabaseConnection, UnsupportedDatabaseException
 from modin.pandas.io import from_arrow, to_pandas
 from modin.test.test_utils import warns_that_defaulting_to_pandas