Skip to content

Commit

Permalink
use new configs module tfor central config validation point
Browse files Browse the repository at this point in the history
  • Loading branch information
rjzamora committed Jul 17, 2024
1 parent fec8038 commit 1bafba8
Show file tree
Hide file tree
Showing 4 changed files with 14 additions and 59 deletions.
55 changes: 0 additions & 55 deletions merlin/__init__.py

This file was deleted.

4 changes: 3 additions & 1 deletion merlin/core/__init__.py
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
#
# Copyright (c) 2022, NVIDIA CORPORATION.
# Copyright (c) 2022-2024, NVIDIA CORPORATION.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
Expand All @@ -14,6 +14,8 @@
# limitations under the License.
#

from merlin.config import validate_dask_configs
from merlin.core import _version

__version__ = _version.get_versions()["version"]
validate_dask_configs()
6 changes: 5 additions & 1 deletion merlin/dag/__init__.py
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
#
# Copyright (c) 2022, NVIDIA CORPORATION.
# Copyright (c) 2022-2024, NVIDIA CORPORATION.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
Expand All @@ -15,6 +15,10 @@
#
# flake8: noqa

from merlin.config import validate_dask_configs

validate_dask_configs()

from merlin.dag.graph import Graph
from merlin.dag.node import Node, iter_nodes, postorder_iter_nodes, preorder_iter_nodes
from merlin.dag.operator import DataFormats, Operator, Supports
Expand Down
8 changes: 6 additions & 2 deletions merlin/io/__init__.py
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
#
# Copyright (c) 2022, NVIDIA CORPORATION.
# Copyright (c) 2022-2024, NVIDIA CORPORATION.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
Expand All @@ -13,8 +13,12 @@
# See the License for the specific language governing permissions and
# limitations under the License.
#

# flake8: noqa

from merlin.config import validate_dask_configs

validate_dask_configs()

from merlin.io import dataframe_iter, dataset, shuffle
from merlin.io.dataframe_iter import DataFrameIter
from merlin.io.dataset import MERLIN_METADATA_DIR_NAME, Dataset
Expand Down

0 comments on commit 1bafba8

Please sign in to comment.