restore import

Author: parthea

.generator/test_cli.py

@@ -16,6 +16,7 @@
 import pytest
 import json
 import logging
+import os
 import subprocess
 
 from unittest.mock import mock_open, MagicMock