mirror of
https://github.com/clearml/clearml
synced 2025-03-12 14:48:30 +00:00
Fix tensorflow_macos support
This commit is contained in:
parent
6529c0dc11
commit
dd79bd6197
@ -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):
|
||||
|
@ -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
|
||||
|
||||
|
||||
|
@ -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'
|
||||
seaborn
|
||||
|
@ -1,3 +1,4 @@
|
||||
numpy != 1.24.0 # https://github.com/numpy/numpy/issues/22826
|
||||
absl-py>=0.7.1
|
||||
tensorboard>=2.0
|
||||
tensorflow>=2.0
|
||||
|
@ -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
|
||||
|
@ -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'
|
||||
|
Loading…
Reference in New Issue
Block a user