Skip to content

Commit 22898a9

Browse files
committed
move settings to core
1 parent 158fcb7 commit 22898a9

File tree

7 files changed

+8
-11
lines changed

7 files changed

+8
-11
lines changed

src/mdio/converters/mdio.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -8,9 +8,9 @@
88
import numpy as np
99
from tqdm.dask import TqdmCallback
1010

11-
from mdio.api._environ import MDIOSettings
1211
from mdio.api.io import _normalize_path
1312
from mdio.api.io import open_mdio
13+
from mdio.core.settings import MDIOSettings
1414
from mdio.segy.blocked_io import to_segy
1515
from mdio.segy.creation import concat_files
1616
from mdio.segy.creation import mdio_spec_to_segy

src/mdio/converters/segy.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,6 @@
1313
from segy.standards.codes import MeasurementSystem as SegyMeasurementSystem
1414
from segy.standards.fields import binary as binary_header_fields
1515

16-
from mdio.api._environ import MDIOSettings
1716
from mdio.api.io import _normalize_path
1817
from mdio.api.io import to_mdio
1918
from mdio.builder.schemas.chunk_grid import RegularChunkGrid
@@ -32,6 +31,7 @@
3231
from mdio.converters.exceptions import GridTraceSparsityError
3332
from mdio.converters.type_converter import to_structured_type
3433
from mdio.core.grid import Grid
34+
from mdio.core.settings import MDIOSettings
3535
from mdio.core.utils_write import MAX_COORDINATES_BYTES
3636
from mdio.core.utils_write import MAX_SIZE_LIVE_MASK
3737
from mdio.core.utils_write import get_constrained_chunksize

src/mdio/api/_environ.py renamed to src/mdio/core/settings.py

Lines changed: 2 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1,10 +1,10 @@
11
"""Environment variable management for MDIO operations."""
22

33
from psutil import cpu_count
4-
from pydantic import ConfigDict
54
from pydantic import Field
65
from pydantic import field_validator
76
from pydantic_settings import BaseSettings
7+
from pydantic_settings import SettingsConfigDict
88

99

1010
class MDIOSettings(BaseSettings):
@@ -58,10 +58,7 @@ class MDIOSettings(BaseSettings):
5858
alias="MDIO_IGNORE_CHECKS",
5959
)
6060

61-
model_config = ConfigDict(
62-
env_prefix="",
63-
case_sensitive=True,
64-
)
61+
model_config = SettingsConfigDict(case_sensitive=True)
6562

6663
@field_validator("save_segy_file_header", "raw_headers", "ignore_checks", "cloud_native", mode="before")
6764
@classmethod

src/mdio/segy/_workers.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -8,8 +8,8 @@
88
import numpy as np
99
from segy.arrays import HeaderArray
1010

11-
from mdio.api._environ import MDIOSettings
1211
from mdio.api.io import _normalize_storage_options
12+
from mdio.core.settings import MDIOSettings
1313
from mdio.segy._raw_trace_wrapper import SegyFileRawTraceWrapper
1414
from mdio.segy.file import SegyFileArguments
1515
from mdio.segy.file import SegyFileWrapper

src/mdio/segy/blocked_io.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -15,12 +15,12 @@
1515
from tqdm.auto import tqdm
1616
from zarr import open_group as zarr_open_group
1717

18-
from mdio.api._environ import MDIOSettings
1918
from mdio.api.io import _normalize_storage_options
2019
from mdio.builder.schemas.v1.stats import CenteredBinHistogram
2120
from mdio.builder.schemas.v1.stats import SummaryStatistics
2221
from mdio.constants import ZarrFormat
2322
from mdio.core.indexing import ChunkIterator
23+
from mdio.core.settings import MDIOSettings
2424
from mdio.segy._workers import trace_worker
2525
from mdio.segy.creation import SegyPartRecord
2626
from mdio.segy.creation import concat_files

src/mdio/segy/parsers.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,7 @@
1111
import numpy as np
1212
from tqdm.auto import tqdm
1313

14-
from mdio.api._environ import MDIOSettings
14+
from mdio.core.settings import MDIOSettings
1515
from mdio.segy._workers import header_scan_worker
1616

1717
if TYPE_CHECKING:

tests/unit/test_environment.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@
66
import pytest
77
from pydantic import ValidationError
88

9-
from mdio.api._environ import MDIOSettings
9+
from mdio.core.settings import MDIOSettings
1010

1111

1212
class TestEnvironment:

0 commit comments

Comments
 (0)