restore import

Author: parthea

.generator/cli.py

@@ -15,6 +15,7 @@
 import argparse
 import json
 import logging
+import os
 import subprocess
 import sys
 import subprocess

.generator/test_cli.py

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