diff --git a/src/openclpow.py b/src/openclpow.py index b8acbd4b..6bed98a5 100644 --- a/src/openclpow.py +++ b/src/openclpow.py @@ -5,9 +5,14 @@ import logging import os from struct import pack -import paths -from bmconfigparser import BMConfigParser -from state import shutdown +try: + import paths + from bmconfigparser import BMConfigParser + from state import shutdown +except ImportError: + from pybitmessage import paths + from pybitmessage.bmconfigparser import BMConfigParser + from pybitmessage.state import shutdown try: import numpy diff --git a/src/tests/test_openclpow.py b/src/tests/test_openclpow.py index 6cfb5bd2..341beec9 100644 --- a/src/tests/test_openclpow.py +++ b/src/tests/test_openclpow.py @@ -4,11 +4,6 @@ Tests for openclpow module import hashlib import unittest from struct import pack, unpack - -from .common import skip_python3 - -skip_python3() # noqa:E402 - from pybitmessage import openclpow