@@ -43,14 +43,19 @@ torch = { version = ">=2.0.0", optional = true }
4343torchvision = { version = " >=0.15" , optional = true }
4444# TensorFlow models
4545tensorflow = [
46- { version = " ^2.7.0,< =2.10" , optional = true , platform = " win32" },
46+ { version = " = =2.10" , optional = true , platform = " win32" },
4747 { version = " ^2.7.0,<=2.12" , optional = true , platform = " linux" },
4848]
4949tensorflow-macos = { version = " ^2.7.0,<=2.12" , optional = true , markers = " sys_platform == 'darwin'" }
5050tensorflow-metal = { version = " <1.3.0" , optional = true , markers = " sys_platform == 'darwin'" }
51+ tensorflow-io-gcs-filesystem = [
52+ { version = " ==0.32" , optional = true , platform = " win32" },
53+ { version = " *" , optional = true , platform = " linux" },
54+ { version = " *" , optional = true , markers = " sys_platform == 'darwin'" }
55+ ]
5156
5257[tool .poetry .extras ]
53- tf = [ " tensorflow" , " tensorflow-macos" , " tensorflow-metal" ]
58+ tf = [ " tensorflow" , " tensorflow-macos" , " tensorflow-metal" , " tensorflow-io-gcs-filesystem " ]
5459pytorch = [" scipy" , " timm" , " torch" , " torchvision" ]
5560
5661[tool .poetry .group .dev .dependencies ]
0 commit comments