fix set_geometry

VROOM-Project · Apr 19, 2024 · bd5f0f3 · bd5f0f3
1 parent df29bf0
commit bd5f0f3
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 5 deletions.
diff --git a/README.rst b/README.rst
@@ -53,8 +53,6 @@ Basic usage
   ...                           vroom.Job(1616, location=2),
   ...                           vroom.Job(1717, location=3)])
 
-  >>> problem_instance.set_geometry()
-
   >>> solution = problem_instance.solve(exploration_level=5, nb_threads=4)
 
   >>> solution.summary.cost
@@ -65,7 +63,7 @@ Basic usage
          'waiting_time', 'location_index', 'id', 'description'],
         dtype='object')
 
-  >>> solution.routes[["vehicle_id", "type", "arrival", "location_index", "id", "distance"]]
+  >>> solution.routes[["vehicle_id", "type", "arrival", "location_index", "id"]]
      vehicle_id   type  arrival  location_index    id
   0          47  start        0               0  <NA>
   1          47    job     2104               1  1515

diff --git a/src/vroom/input/input.py b/src/vroom/input/input.py
@@ -107,15 +107,15 @@ def from_json(
         if geometry is None:
             geometry = servers is not None
         if geometry:
-            self._set_geometry()
+            self._set_geometry(True)
         instance = Input(servers=servers, router=router)
         with open(filepath) as handle:
             instance._from_json(handle.read(), geometry)
         return instance
 
     def set_geometry(self):
         self._geometry = True
-        return self._set_geometry()
+        return self._set_geometry(True)
 
     def set_amount_size(self, *amount_sizes: int) -> None:
         """Add amount sizes."""