flavor_matcher: add missing imports

rackerlabs · Nov 20, 2024 · 01e05f2 · 01e05f2
1 parent f56e7a6
commit 01e05f2
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/python/understack-flavor-matcher/tests/test_flavor_spec.py b/python/understack-flavor-matcher/tests/test_flavor_spec.py
@@ -1,4 +1,6 @@
-from unittest.mock import mock_open, patch
+import os
+from unittest.mock import mock_open
+from unittest.mock import patch
 
 import pytest
 from flavor_matcher.flavor_spec import FlavorSpec

diff --git a/python/understack-workflows/understack_workflows/flavor_detect.py b/python/understack-workflows/understack_workflows/flavor_detect.py
@@ -2,6 +2,7 @@
 from flavor_matcher.matcher import FlavorSpec
 from flavor_matcher.matcher import Matcher
 
+from understack_workflows import bmc_disk
 from understack_workflows.bmc import Bmc
 from understack_workflows.bmc_chassis_info import ChassisInfo
 from understack_workflows.helpers import setup_logger