Merge pull request DavHau#541 from Preemo-Inc/deepspeed-mii.remove-as…

…yncio Add deepspeed-mii.remove-asyncio to fixes
leo60228 · Mar 25, 2023 · a343c27 · a343c27
2 parents f60b983 + 19c6585
commit a343c27
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/mach_nix/fixes.nix b/mach_nix/fixes.nix
@@ -129,4 +129,11 @@ rec {
     patchPhase = "";
   };
 
+  deepspeed-mii.remove-asyncio = {
+    _cond = ({ pyver, ... }:
+      # asyncio becomes a built-in library since Python 3.4
+      comp_ver pyver ">=" "3.4");
+    propagatedBuildInputs.mod =
+      builtins.filter (input: input.pname != "asyncio");
+  };
 }
diff --git a/mach_nix/nix/lib.nix b/mach_nix/nix/lib.nix
@@ -193,7 +193,7 @@ rec {
     let
       provider = if hasAttr "provider" oa.passthru then oa.passthru.provider else "nixpkgs";
     in
-      condition { prov = provider; ver = oa.version; pyver = oa.pythonModule.version; };
+      condition { prov = provider; ver = oa.version; pyver = oa.passthru.pythonModule.version; };
 
     extract = python: src: fail_msg:
     let