REFACTOR-modin-project#6889: Define 'all' for 'envvars.py'

Signed-off-by: Anatoly Myachev <anatoly.myachev@intel.com>
anmyachev · Jan 29, 2024 · f2fa1ca · f2fa1ca
1 parent fb4a19f
commit f2fa1ca
Show file tree

Hide file tree

Showing 2 changed files with 51 additions and 1 deletion.
diff --git a/modin/config/envvars.py b/modin/config/envvars.py
@@ -863,3 +863,53 @@ def _check_vars() -> None:
 
 
 _check_vars()
+
+
+__all__ = [
+    "EnvironmentVariable",
+    # 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",
+]
diff --git a/modin/core/execution/ray/common/utils.py b/modin/core/execution/ray/common/utils.py
@@ -35,8 +35,8 @@
     RayRedisAddress,
     RayRedisPassword,
     StorageFormat,
-    ValueSource,
 )
+from modin.config.pubsub import ValueSource
 from modin.core.execution.utils import set_env
 from modin.error_message import ErrorMessage