Skip to content

Commit 1273c23

Browse files
lu-wang-gtflite-support-robot
authored andcommitted
Remove Task Library from Pypi __init__ on Windows.
PiperOrigin-RevId: 446271636
1 parent 6d70fc9 commit 1273c23

File tree

1 file changed

+6
-1
lines changed

1 file changed

+6
-1
lines changed

tensorflow_lite_support/tools/pip_package/tflite_support.__init__.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -33,8 +33,13 @@
3333
# In pip build, this file will be renamed as tflite_support/__init__.py.
3434

3535
import flatbuffers
36+
import platform
37+
3638
from tensorflow_lite_support.metadata import metadata_schema_py_generated
3739
from tensorflow_lite_support.metadata import schema_py_generated
3840
from tensorflow_lite_support.metadata.python import metadata
3941
from tflite_support import metadata_writers
40-
from tflite_support import task
42+
43+
if platform.system() != 'Windows':
44+
# Task Library is not supported on Windows yet.
45+
from tflite_support import task

0 commit comments

Comments
 (0)