From 82331b7936d8142720bf9d0d5f0964a8565e504e Mon Sep 17 00:00:00 2001 From: Jimmy Shen <14003693+jmmshn@users.noreply.github.com> Date: Fri, 21 Feb 2025 09:15:12 -0800 Subject: [PATCH] config update (#967) Co-authored-by: Jimmy Shen --- mp_api/client/core/settings.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/mp_api/client/core/settings.py b/mp_api/client/core/settings.py index 67455204..91dfff89 100644 --- 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_")