Move TestType to new.internal

src/neptune/new/attributes/series/file_series.py

@@ -22,7 +22,6 @@
     Optional,
 )
 
-from neptune.internal.file_types import FileType
 from neptune.new.attributes.series.series import Series
 from neptune.new.exceptions import (
     FileNotFound,
@@ -34,6 +33,7 @@
     LogImages,
     Operation,
 )
+from neptune.new.internal.types.file_types import FileType
 from neptune.new.internal.utils import base64_encode
 from neptune.new.internal.utils.iteration import get_batches
 from neptune.new.internal.utils.limits import image_size_exceeds_limit_for_logging

src/neptune/new/internal/backends/neptune_backend_mock.py

@@ -35,7 +35,6 @@
     InternalClientError,
     NeptuneException,
 )
-from neptune.internal.file_types import FileType
 from neptune.new.exceptions import (
     ContainerUUIDNotFound,
     MetadataInconsistency,
@@ -105,6 +104,7 @@
     UploadFileSet,
 )
 from neptune.new.internal.operation_visitor import OperationVisitor
+from neptune.new.internal.types.file_types import FileType
 from neptune.new.internal.utils import base64_decode
 from neptune.new.internal.utils.generic_attribute_mapper import NoValue
 from neptune.new.internal.utils.paths import path_to_str

src/neptune/new/internal/operation.py

@@ -30,9 +30,9 @@
 )
 
 from neptune.common.exceptions import InternalClientError
-from neptune.internal.file_types import FileType
 from neptune.new.exceptions import MalformedOperation
 from neptune.new.internal.container_type import ContainerType
+from neptune.new.internal.types.file_types import FileType
 from neptune.new.types.atoms.file import File
 
 if TYPE_CHECKING:

src/neptune/new/internal/types/__init__.py

@@ -0,0 +1,15 @@
+#
+# Copyright (c) 2022, Neptune Labs Sp. z o.o.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# 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.
+#

src/neptune/new/types/atoms/file.py

@@ -25,7 +25,7 @@
     Union,
 )
 
-from neptune.internal.file_types import (
+from neptune.new.internal.types.file_types import (
     FileComposite,
     InMemoryComposite,
     LocalFileComposite,

tests/neptune/new/attributes/atoms/test_file.py

@@ -26,7 +26,6 @@
 
 from e2e_tests.utils import tmp_context
 from neptune.common.utils import IS_WINDOWS
-from neptune.internal.file_types import FileType
 from neptune.new.attributes.atoms.file import (
     File,
     FileVal,
@@ -39,6 +38,7 @@
     UploadFile,
     UploadFileSet,
 )
+from neptune.new.internal.types.file_types import FileType
 from tests.neptune.new.attributes.test_attribute_base import TestAttributeBase
 
 

tests/neptune/new/types/atoms/test_file.py

@@ -25,14 +25,14 @@
 from PIL import Image
 
 from e2e_tests.utils import tmp_context
-from neptune.internal.file_types import (
-    FileType,
-    InMemoryComposite,
-)
 from neptune.new.exceptions import (
     NeptuneException,
     StreamAlreadyUsedException,
 )
+from neptune.new.internal.types.file_types import (
+    FileType,
+    InMemoryComposite,
+)
 from neptune.new.internal.utils.images import _get_pil_image_data
 from neptune.new.types import File
 from tests.neptune.new.attributes.test_attribute_base import TestAttributeBase