diff --git a/nemo_curator/scripts/classifiers/fineweb_mixtral_edu_classifier_inference.py b/nemo_curator/scripts/classifiers/fineweb_mixtral_edu_classifier_inference.py index 582ec4c5..756584f3 100644 --- a/nemo_curator/scripts/classifiers/fineweb_mixtral_edu_classifier_inference.py +++ b/nemo_curator/scripts/classifiers/fineweb_mixtral_edu_classifier_inference.py @@ -1,4 +1,4 @@ -# Copyright (c) 2024, NVIDIA CORPORATION. All rights reserved. +# Copyright (c) 2025, NVIDIA CORPORATION. All rights reserved. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/nemo_curator/scripts/classifiers/fineweb_nemotron_edu_classifier_inference.py b/nemo_curator/scripts/classifiers/fineweb_nemotron_edu_classifier_inference.py index 112453a2..d58867ce 100644 --- a/nemo_curator/scripts/classifiers/fineweb_nemotron_edu_classifier_inference.py +++ b/nemo_curator/scripts/classifiers/fineweb_nemotron_edu_classifier_inference.py @@ -1,4 +1,4 @@ -# Copyright (c) 2024, NVIDIA CORPORATION. All rights reserved. +# Copyright (c) 2025, NVIDIA CORPORATION. All rights reserved. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/pyproject.toml b/pyproject.toml index 56e0fd9e..ec270713 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -1,4 +1,4 @@ -# Copyright (c) 2024, NVIDIA CORPORATION. All rights reserved. +# Copyright (c) 2025, NVIDIA CORPORATION. All rights reserved. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/tests/test_classifiers.py b/tests/test_classifiers.py index d6d2852c..81b1112e 100644 --- a/tests/test_classifiers.py +++ b/tests/test_classifiers.py @@ -1,4 +1,4 @@ -# Copyright (c) 2024, NVIDIA CORPORATION. All rights reserved. +# Copyright (c) 2025, NVIDIA CORPORATION. All rights reserved. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. @@ -139,9 +139,6 @@ def test_fineweb_edu_classifier(gpu_client, domain_dataset): assert result_pred.equals(expected_pred) -@pytest.mark.skip( - reason="Skipping until https://huggingface.co/nvidia/nemocurator-fineweb-mixtral-edu-classifier is published" -) @pytest.mark.gpu def test_fineweb_mixtral_classifier(gpu_client, domain_dataset): from nemo_curator.classifiers import FineWebMixtralEduClassifier @@ -155,9 +152,6 @@ def test_fineweb_mixtral_classifier(gpu_client, domain_dataset): assert result_pred.equals(expected_pred) -@pytest.mark.skip( - reason="Skipping until https://huggingface.co/nvidia/nemocurator-fineweb-nemotron-4-edu-classifier is published" -) @pytest.mark.gpu def test_fineweb_nemotron_classifier(gpu_client, domain_dataset): from nemo_curator.classifiers import FineWebNemotronEduClassifier