Merge branch 'main' of github.com:ewfuentes/robot

experimental/beacon_dist/utils_test.py

@@ -123,7 +123,7 @@ def test_valid_configuration_loss_bad_prediction_yields_nonzero_loss(self):
         loss = valid_configuration_loss(batch.class_label, query, model_output)
 
         # Verification
-        self.assertGreater(loss, 0.0)
+        self.assertGreater(loss.item(), 0.0)
 
     def test_valid_configuration_loss_missing_exclusive_keypoints_are_valid(self):
         # Setup
@@ -136,7 +136,7 @@ def test_valid_configuration_loss_missing_exclusive_keypoints_are_valid(self):
         loss = valid_configuration_loss(batch.class_label, query, model_output)
 
         # Verification
-        self.assertAlmostEqual(loss, 0.0)
+        self.assertAlmostEqual(loss.item(), 0.0)
 
     def test_query_from_class_samples(self):
         # Setup

third_party/python/BUILD

@@ -6,15 +6,15 @@ load("@pip//:requirements.bzl", "requirement")
 
 compile_pip_requirements_3_8(
   name = "requirements_3_8",
-  requirements_in=":requirements.in",
+  requirements_in=":requirements_3_8.in",
   requirements_txt=":requirements_3_8.txt",
   extra_args=["--allow-unsafe"],
   timeout="eternal",
 )
 
 compile_pip_requirements_3_10(
   name = "requirements_3_10",
-  requirements_in=":requirements.in",
+  requirements_in=":requirements_3_10.in",
   requirements_txt=":requirements_3_10.txt",
   extra_args=["--allow-unsafe"],
   timeout="eternal",

third_party/python/requirements_3_10.in

@@ -0,0 +1,25 @@
+sympy==1.13.1
+numpy==1.24.4
+jupyterlab==4.0.8
+matplotlib==3.7.5
+Pillow==10.4.0
+pandas==1.5.3
+scikit-learn==1.2.0
+tqdm==4.67.1
+opencv-python==4.7.0.72
+Wand==0.6.11
+torch==2.5.1
+setuptools==67.6.1
+ipdb==0.13.13
+ipympl==0.9.3
+beautifulsoup4==4.12.2
+line_profiler==4.0.3
+PyYAML==6.0
+PyGObject==3.44.1
+click==8.1.7
+spatialmath-python==1.1.8
+reportlab==4.1.0
+seaborn==0.13.2
+depthai==2.28.0.0
+rosbags==0.9.23
+sam-2 @ git+https://github.com/IDEA-Research/Grounded-SAM-2.git@dd4c5141b75e4838dd486c64f773c43b4db3a07b