diff --git a/src/unitgrade/framework.py b/src/unitgrade/framework.py index ce06d0d882c7eac13d481a5deed0a46f05b8cd1f..1f1a29914367e5ef0b6195cc7f098e027c21b8fc 100644 --- a/src/unitgrade/framework.py +++ b/src/unitgrade/framework.py @@ -8,6 +8,7 @@ import colorama import unittest import time import textwrap + from unitgrade.runners import UTextResult from unitgrade.utils import gprint, Capturing2, Capturing colorama.init(autoreset=True) # auto resets your settings after every output