From aae5f9e0405d3364bbd5cc875b80e325f71a79fc Mon Sep 17 00:00:00 2001 From: Varun Agrawal Date: Sun, 22 Sep 2024 23:15:00 -0400 Subject: [PATCH] fix numpy deprecation --- python/gtsam/tests/test_NonlinearOptimizer.py | 17 +++++++++-------- 1 file changed, 9 insertions(+), 8 deletions(-) diff --git a/python/gtsam/tests/test_NonlinearOptimizer.py b/python/gtsam/tests/test_NonlinearOptimizer.py index a47c5ad62..1b4209509 100644 --- a/python/gtsam/tests/test_NonlinearOptimizer.py +++ b/python/gtsam/tests/test_NonlinearOptimizer.py @@ -14,15 +14,16 @@ from __future__ import print_function import unittest -import gtsam -from gtsam import ( - DoglegOptimizer, DoglegParams, DummyPreconditionerParameters, GaussNewtonOptimizer, - GaussNewtonParams, GncLMParams, GncLossType, GncLMOptimizer, LevenbergMarquardtOptimizer, - LevenbergMarquardtParams, NonlinearFactorGraph, Ordering, PCGSolverParameters, Point2, - PriorFactorPoint2, Values -) from gtsam.utils.test_case import GtsamTestCase +import gtsam +from gtsam import (DoglegOptimizer, DoglegParams, + DummyPreconditionerParameters, GaussNewtonOptimizer, + GaussNewtonParams, GncLMOptimizer, GncLMParams, GncLossType, + LevenbergMarquardtOptimizer, LevenbergMarquardtParams, + NonlinearFactorGraph, Ordering, PCGSolverParameters, Point2, + PriorFactorPoint2, Values) + KEY1 = 1 KEY2 = 2 @@ -136,7 +137,7 @@ class TestScenario(GtsamTestCase): # Test optimizer params optimizer = GncLMOptimizer(self.fg, self.initial_values, params) for ict_factor in (0.9, 1.1): - new_ict = ict_factor * optimizer.getInlierCostThresholds() + new_ict = ict_factor * optimizer.getInlierCostThresholds().item() optimizer.setInlierCostThresholds(new_ict) self.assertAlmostEqual(optimizer.getInlierCostThresholds(), new_ict) for w_factor in (0.8, 0.9):