fix: pytorch from nixpkgs not detected as torch (fixes #148)
This commit is contained in:
parent
8be56fc14d
commit
52fc618e77
1 changed files with 6 additions and 1 deletions
|
@ -22,6 +22,10 @@ let
|
|||
});
|
||||
};
|
||||
|
||||
nameMap = {
|
||||
pytorch = "torch";
|
||||
};
|
||||
|
||||
py = python.override { packageOverrides = mergeOverrides ( overrides ++ [ fetchPypiPnamePassthruOverride ] ); };
|
||||
in
|
||||
|
||||
|
@ -46,7 +50,8 @@ let
|
|||
if hasAttrByPath ["${attrname}" "src" "pname"] python.pkgs then
|
||||
python.pkgs."${attrname}".src.pname
|
||||
else if hasAttrByPath ["${attrname}" "pname"] python.pkgs then
|
||||
python.pkgs."${attrname}".pname
|
||||
let pname = python.pkgs."${attrname}".pname; in
|
||||
if nameMap ? "${pname}" then nameMap."${pname}" else pname
|
||||
else ""
|
||||
);
|
||||
in
|
||||
|
|
Loading…
Reference in a new issue