summaryrefslogtreecommitdiff
path: root/gr-fec/python/fec/fec_test.py
diff options
context:
space:
mode:
Diffstat (limited to 'gr-fec/python/fec/fec_test.py')
-rw-r--r--gr-fec/python/fec/fec_test.py8
1 files changed, 7 insertions, 1 deletions
diff --git a/gr-fec/python/fec/fec_test.py b/gr-fec/python/fec/fec_test.py
index 0a7524af74..6466a0bcb4 100644
--- a/gr-fec/python/fec/fec_test.py
+++ b/gr-fec/python/fec/fec_test.py
@@ -21,8 +21,14 @@
#
from gnuradio.fec.bitflip import read_bitlist
-from gnuradio import gr, blocks, analog, digital
+from gnuradio import gr, blocks, analog
import math
+import sys
+
+if sys.modules.has_key("gnuradio.digital"):
+ digital = sys.modules["gnuradio.digital"]
+else:
+ from gnuradio import digital
from extended_encoder import extended_encoder
from extended_decoder import extended_decoder