Fix tensorflow_macos support

This commit is contained in:
allegroai 2022-12-22 21:54:13 +02:00
parent 6529c0dc11
commit dd79bd6197
6 changed files with 13 additions and 3 deletions

View File

@ -14,7 +14,7 @@ except ImportError:
from ..config import running_remotely, get_remote_task_id
from .frameworks import _patched_call # noqa
from ..utilities.proxy_object import verify_basic_type
from ..utilities.proxy_object import verify_basic_type, flatten_dictionary
class PatchJsonArgParse(object):

View File

@ -360,6 +360,12 @@ def get_installed_pkgs_detail():
if 'tensorflow_gpu' in mapping:
mapping['tensorflow'] = mapping['tensorflow_gpu']
# HACK: prefer tensorflow_macos over tensorflow
if 'tensorflow_macos' in mapping:
p = mapping.pop('tensorflow_macos', None)
if p and isinstance(p, (tuple, list)) and len(p) == 2:
mapping['tensorflow'] = ('tensorflow', p[1])
return mapping

View File

@ -1,4 +1,5 @@
numpy != 1.24.0 # https://github.com/numpy/numpy/issues/22826
matplotlib >= 3.1.1 ; python_version >= '3.6'
matplotlib >= 2.2.4 ; python_version < '3.6'
seaborn
clearml
clearml

View File

@ -1,4 +1,5 @@
numpy != 1.24.0 # https://github.com/numpy/numpy/issues/22826
absl-py>=0.7.1
tensorboard>=2.0
tensorflow>=2.0
clearml
clearml

View File

@ -1,3 +1,4 @@
numpy != 1.24.0 # https://github.com/numpy/numpy/issues/22826
matplotlib >= 3.1.1 ; python_version >= '3.6'
matplotlib >= 2.2.4 ; python_version < '3.6'
scikit-learn

View File

@ -1,3 +1,4 @@
numpy != 1.24.0 # https://github.com/numpy/numpy/issues/22826
bokeh>=1.4.0
matplotlib >= 3.1.1 ; python_version >= '3.6'
matplotlib >= 2.2.4 ; python_version < '3.6'