diff --git a/nemo_curator/services/nemo_client.py b/nemo_curator/services/nemo_client.py index f83ba624..d84c72f1 100644 --- a/nemo_curator/services/nemo_client.py +++ b/nemo_curator/services/nemo_client.py @@ -11,16 +11,15 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. +from __future__ import annotations + import warnings from typing import Iterable, List, Optional, Union from nemo_curator.services.conversation_formatter import ConversationFormatter -from nemo_curator.utils.import_utils import safe_import_from from .model_client import AsyncLLMClient, LLMClient -NemoQueryLLM = safe_import_from("nemo.deploy.nlp", "NemoQueryLLM") - class NemoDeployClient(LLMClient): """ diff --git a/setup.py b/setup.py index 185fb3af..c3a16271 100644 --- a/setup.py +++ b/setup.py @@ -45,7 +45,7 @@ "charset_normalizer>=3.1.0", "awscli>=1.22.55", "fasttext==0.9.2", - "pycld2==0.41", + "pycld2", "justext==3.0.1", "resiliparse", "ftfy==6.1.1",