Make contracts import path-independent from the caller

INET-Complexity · Dec 20, 2018 · 25a7003 · 25a7003
1 parent 15cf5e1
commit 25a7003
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/agents/metainsuranceorg.py b/agents/metainsuranceorg.py
@@ -3,8 +3,7 @@
 import numpy as np
 import scipy.stats
 import copy
-from contracts.insurancecontract import InsuranceContract
-from contracts.reinsurancecontract import ReinsuranceContract
+from contracts import InsuranceContract, ReinsuranceContract
 from riskmodel import RiskModel
 import sys, pdb
 import uuid

diff --git a/contracts/__init__.py b/contracts/__init__.py
@@ -0,0 +1,2 @@
+from .insurancecontract import InsuranceContract
+from .reinsurancecontract import ReinsuranceContract