diff --git a/risk/__init__.py b/risk/__init__.py index 57e7ee1..66cadf2 100644 --- a/risk/__init__.py +++ b/risk/__init__.py @@ -66,4 +66,8 @@ Basic examples that illustrate usage of the the framework are in the notebook fo """ -from risk import deid +import sys +if sys.version_info.major == 2: + from risk import deid +else: + from risk.risk import deid