config update (#967)

Co-authored-by: Jimmy Shen <v-jimshen@microsoft.com>
materialsproject · Feb 21, 2025 · 82331b7 · 82331b7
1 parent f7a89d4
commit 82331b7
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/mp_api/client/core/settings.py b/mp_api/client/core/settings.py
@@ -3,7 +3,7 @@
 from typing import List
 
 from pydantic import Field
-from pydantic_settings import BaseSettings
+from pydantic_settings import SettingsConfigDict, BaseSettings
 from pymatgen.core import _load_pmg_settings
 
 from mp_api.client import __file__ as root_dir
@@ -26,6 +26,8 @@ class MAPIClientSettings(BaseSettings):
     python module.
     """
 
+    model_config = ConfigDict(env_prefix="MPRESTER_")
+
     TEST_FILES: str = Field(
         os.path.join(os.path.dirname(os.path.abspath(root_dir)), "../../test_files"),
         description="Directory with test files",
@@ -86,6 +88,4 @@ class MAPIClientSettings(BaseSettings):
     MAX_LIST_LENGTH: int = Field(
         _MAX_LIST_LENGTH, description="Maximum length of query parameter list"
     )
-
-    class Config:
-        env_prefix = "MPRESTER_"
+    model_config = SettingsConfigDict(env_prefix="MPRESTER_")