Update cognitivess_model/__init__.py

cognitivess_model/__init__.py

@@ -1,4 +1,4 @@
-# Copyright 2023 Cognitivess AI and The HuggingFace Inc. team. All rights reserved.
+# Copyright 2022 Cognitivess and The HuggingFace Inc. team. 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.
@@ -17,7 +17,8 @@ from ...utils import (
     OptionalDependencyNotAvailable,
     _LazyModule,
     is_flax_available,
-    is_tf_available,
+    is_sentencepiece_available,
+    is_tokenizers_available,
     is_torch_available,
 )
 
@@ -26,6 +27,21 @@ _import_structure = {
     "configuration_Cognitivess": ["CognitivessConfig"],
 }
 
+try:
+    if not is_sentencepiece_available():
+        raise OptionalDependencyNotAvailable()
+except OptionalDependencyNotAvailable:
+    pass
+else:
+    _import_structure["tokenization_Cognitivess"] = ["CognitivessTokenizer"]
+
+try:
+    if not is_tokenizers_available():
+        raise OptionalDependencyNotAvailable()
+except OptionalDependencyNotAvailable:
+    pass
+else:
+    _import_structure["tokenization_Cognitivess_fast"] = ["CognitivessTokenizerFast"]
 
 try:
     if not is_torch_available():
@@ -38,6 +54,7 @@ else:
         "CognitivessModel",
         "CognitivessPreTrainedModel",
         "CognitivessForSequenceClassification",
+        "CognitivessForQuestionAnswering",
         "CognitivessForTokenClassification",
     ]
 
@@ -47,29 +64,28 @@ try:
 except OptionalDependencyNotAvailable:
     pass
 else:
-    _import_structure["modeling_flax_Cognitivess"] = [
-        "FlaxCognitivessForCausalLM",
-        "FlaxCognitivessModel",
-        "FlaxCognitivessPreTrainedModel",
-    ]
-
-try:
-    if not is_tf_available():
-        raise OptionalDependencyNotAvailable()
-except OptionalDependencyNotAvailable:
-    pass
-else:
-    _import_structure["modeling_tf_Cognitivess"] = [
-        "TFCognitivessModel",
-        "TFCognitivessForCausalLM",
-        "TFCognitivessForSequenceClassification",
-        "TFCognitivessPreTrainedModel",
-    ]
+    _import_structure["modeling_flax_Cognitivess"] = ["FlaxCognitivessForCausalLM", "FlaxCognitivessModel", "FlaxCognitivessPreTrainedModel"]
 
 
 if TYPE_CHECKING:
     from .configuration_Cognitivess import CognitivessConfig
 
+    try:
+        if not is_sentencepiece_available():
+            raise OptionalDependencyNotAvailable()
+    except OptionalDependencyNotAvailable:
+        pass
+    else:
+        from .tokenization_Cognitivess import CognitivessTokenizer
+
+    try:
+        if not is_tokenizers_available():
+            raise OptionalDependencyNotAvailable()
+    except OptionalDependencyNotAvailable:
+        pass
+    else:
+        from .tokenization_Cognitivess_fast import CognitivessTokenizerFast
+
     try:
         if not is_torch_available():
             raise OptionalDependencyNotAvailable()
@@ -78,6 +94,7 @@ if TYPE_CHECKING:
     else:
         from .modeling_Cognitivess import (
             CognitivessForCausalLM,
+            CognitivessForQuestionAnswering,
             CognitivessForSequenceClassification,
             CognitivessForTokenClassification,
             CognitivessModel,
@@ -90,24 +107,7 @@ if TYPE_CHECKING:
     except OptionalDependencyNotAvailable:
         pass
     else:
-        from .modeling_flax_Cognitivess import (
-            FlaxCognitivessForCausalLM,
-            FlaxCognitivessModel,
-            FlaxCognitivessPreTrainedModel,
-        )
-
-    try:
-        if not is_tf_available():
-            raise OptionalDependencyNotAvailable()
-    except OptionalDependencyNotAvailable:
-        pass
-    else:
-        from .modeling_tf_Cognitivess import (
-            TFCognitivessForCausalLM,
-            TFCognitivessForSequenceClassification,
-            TFCognitivessModel,
-            TFCognitivessPreTrainedModel,
-        )
+        from .modeling_flax_Cognitivess import FlaxCognitivessForCausalLM, FlaxCognitivessModel, FlaxCognitivessPreTrainedModel
 
 
 else: