Checkpoints callbacks fixes

decloud/core/dataset.py

@@ -21,6 +21,7 @@ FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
 DEALINGS IN THE SOFTWARE.
 """
 """Dataset classes"""
+import os
 import abc
 import collections.abc
 import json