fix: Config not loading environment variables for config (#2136)

Author: LeonLuttenberger

awswrangler/_config.py

@@ -144,12 +144,12 @@ def to_pandas(self) -> pd.DataFrame:
     def _load_config(self, name: str) -> bool:
         if _CONFIG_ARGS[name].loaded:
             self._set_config_value(key=name, value=_CONFIG_ARGS[name].default)
-            return True
 
         env_var: Optional[str] = os.getenv(f"WR_{name.upper()}")
         if env_var is not None:
             self._set_config_value(key=name, value=env_var)
             return True
+
         return False
 
     def _set_config_value(self, key: str, value: Any) -> None:

tests/test_config.py

@@ -258,3 +258,13 @@ def test_athena_wait_delay_config_override(wr: ModuleType, glue_database: str) -
         mock_wait_query.assert_called_once()
 
         assert mock_wait_query.call_args[1]["athena_query_wait_polling_delay"] == polling_delay_argument
+
+
+@pytest.mark.parametrize("suppress_warnings", [False, True])
+def test_load_from_env_variable(wr: ModuleType, suppress_warnings: bool) -> None:
+    env_variable_value = "1" if suppress_warnings else ""
+
+    with patch.dict(os.environ, {"WR_SUPPRESS_WARNINGS": env_variable_value}):
+        wr.config.reset()
+
+        assert wr.config.suppress_warnings == suppress_warnings