diff options
Diffstat (limited to 'pkgs/servers/home-assistant/parse-requirements.py')
-rwxr-xr-x | pkgs/servers/home-assistant/parse-requirements.py | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/pkgs/servers/home-assistant/parse-requirements.py b/pkgs/servers/home-assistant/parse-requirements.py index a2cf2d0386a..647e4513de4 100755 --- a/pkgs/servers/home-assistant/parse-requirements.py +++ b/pkgs/servers/home-assistant/parse-requirements.py @@ -33,7 +33,9 @@ PKG_SET = 'python3Packages' # the following can be used to choose one of them PKG_PREFERENCES = { # Use python3Packages.youtube-dl-light instead of python3Packages.youtube-dl - 'youtube-dl': 'youtube-dl-light' + 'youtube-dl': 'youtube-dl-light', + 'tensorflow-bin': 'tensorflow', + 'tensorflowWithoutCuda': 'tensorflow' } def get_version(): |