add pypy

Defelo · Dec 18, 2024 · 94aee51 · 94aee51
1 parent 6deda94
commit 94aee51
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 1 deletion.
diff --git a/Python/lib/__init__.py b/Python/lib/__init__.py
@@ -29,7 +29,10 @@
 from heapq import heapify, heappop, heappush
 from pathlib import Path
 
-import numpy as np
+try:
+    import numpy as np
+except:
+    pass
 
 YEAR: int | None = None
 DAY: int | None = None
diff --git a/flake.nix b/flake.nix
@@ -36,6 +36,14 @@
                 sympy
               ]);
 
+            pypy = pkgs.pypy310.withPackages (p:
+              with p; [
+                # z3
+                # numpy
+                networkx
+                sympy
+              ]);
+
             downloadInput = pkgs.stdenvNoCC.mkDerivation {
               name = "aoc-download-input";
               dontUnpack = true;
@@ -111,6 +119,7 @@
       fetchRanks,
       live,
       python,
+      pypy,
       ...
     }: {
       default = pkgs.mkShell {
@@ -129,6 +138,7 @@
 
           # Python
           python
+          pypy
 
           # Haskell
           (haskellPackages.ghcWithPackages (p: with p; [split regex-tdfa]))