Fix imports

leonlan · Sep 22, 2023 · 6016f72 · 6016f72
1 parent 08b1c14
commit 6016f72
Show file tree

Hide file tree

Showing 5 changed files with 5 additions and 5 deletions.
diff --git a/benchmark.py b/benchmark.py
@@ -13,7 +13,7 @@
 from ddwp.read import read
 from ddwp.sampling import SAMPLING_METHODS
 from ddwp.static_solvers import default_solver
-from pyvrp import CostEvaluator
+from pyvrp._pyvrp import CostEvaluator
 
 
 def parse_args():

diff --git a/ddwp/static_solvers/default_solver.py b/ddwp/static_solvers/default_solver.py
@@ -1,5 +1,5 @@
 from ddwp.VrpInstance import VrpInstance
-from pyvrp import Model, Result
+from pyvrp._pyvrp import Model, Result
 from pyvrp.stop import MaxRuntime
 
 from .instance2data import instance2data

diff --git a/ddwp/static_solvers/instance2data.py b/ddwp/static_solvers/instance2data.py
@@ -3,7 +3,7 @@
 import numpy as np
 
 from ddwp.VrpInstance import VrpInstance
-from pyvrp import Client, ProblemData, VehicleType
+from pyvrp._pyvrp import Client, ProblemData, VehicleType
 
 
 def instance2data(instance: VrpInstance) -> ProblemData:

diff --git a/ddwp/static_solvers/scenario_solver.py b/ddwp/static_solvers/scenario_solver.py
@@ -1,7 +1,7 @@
 import warnings
 
 from ddwp.VrpInstance import VrpInstance
-from pyvrp import (
+from pyvrp._pyvrp import (
     GeneticAlgorithm,
     PenaltyManager,
     Population,

diff --git a/tune/benchmark_static.py b/tune/benchmark_static.py
@@ -13,7 +13,7 @@
     raise ModuleNotFoundError(msg) from exc
 
 import pyvrp.search
-from pyvrp import (
+from pyvrp._pyvrp import (
     GeneticAlgorithm,
     GeneticAlgorithmParams,
     PenaltyManager,