Fixed unit test and added call to Objective.update() in update_vector…

…ization() if batch_size is None.

theseus/core/objective.py

@@ -479,6 +479,8 @@ def _get_batch_size(batch_sizes: Sequence[int]) -> int:
     def update_vectorization(self):
         if self._vectorization_run is None:
             return
+        if self._batch_size is None:
+            self.update()
         self._vectorization_run()
 
     # iterates over cost functions

theseus/core/tests/test_vectorizer.py

@@ -177,7 +177,7 @@ def test_vectorized_error():
             objective.add(th.Difference(s, ws, s_target))
 
         vectorization = th.Vectorize(objective)
-        objective.update()
+        objective.update_vectorization()
 
         assert objective._cost_functions_iterable is vectorization._cost_fn_wrappers
         for w in vectorization._cost_fn_wrappers: