fix python deps

flake.nix

@@ -3,43 +3,70 @@
 
   inputs = {
     nix-ros-overlay.url = "github:lopsided98/nix-ros-overlay/master";
-    nixpkgs.follows = "nix-ros-overlay/nixpkgs";  # IMPORTANT!!!
+    nixpkgs.follows = "nix-ros-overlay/nixpkgs"; # IMPORTANT!!!
   };
 
-  outputs = { self, nix-ros-overlay, nixpkgs }:
+  outputs =
-    nix-ros-overlay.inputs.flake-utils.lib.eachDefaultSystem (system:
+    {
+      self,
+      nix-ros-overlay,
+      nixpkgs,
+    }:
+    nix-ros-overlay.inputs.flake-utils.lib.eachDefaultSystem (
+      system:
       let
         pkgs = import nixpkgs {
           inherit system;
           overlays = [ nix-ros-overlay.overlays.default ];
         };
-      in {
+      in
+      {
         devShells.default = pkgs.mkShell {
           name = "ASTRA Anchor";
           packages = with pkgs; [
             colcon
-            python312Packages.pyserial
+            (python312.withPackages (
-            python312Packages.pygame
+              p: with p; [
-            (with rosPackages.humble; buildEnv {
+                pyserial
-              paths = [
+                pygame
-                ros-core ros2cli ros2run ros2bag rviz2 xacro ament-cmake-core python-cmake-module
+                scipy
-                diff-drive-controller parameter-traits generate-parameter-library
+              ]
-                joint-state-publisher-gui robot-state-publisher
+            ))
-                ros2-control controller-manager
+            (
-                # ros2-controllers nixpkg does not build :(
+              with rosPackages.humble;
-              ];
+              buildEnv {
-            })
+                paths = [
+                  ros-core
+                  ros2cli
+                  ros2run
+                  ros2bag
+                  rviz2
+                  xacro
+                  ament-cmake-core
+                  python-cmake-module
+                  diff-drive-controller
+                  parameter-traits
+                  generate-parameter-library
+                  joint-state-publisher-gui
+                  robot-state-publisher
+                  ros2-control
+                  controller-manager
+                  # ros2-controllers nixpkg does not build :(
+                ];
+              }
+            )
           ];
           shellHook = ''
             [ -d install ] || colcon build --symlink-install
             source install/setup.bash
-            
+
             # Display stuff
             export DISPLAY=''${DISPLAY:-:0}
             export QT_X11_NO_MITSHM=1
           '';
         };
-      });
+      }
+    );
 
   nixConfig = {
     extra-substituters = [ "https://ros.cachix.org" ];