Fix an import

Author: sage-maker

tests/unit/sagemaker/workflow/test_notebook_job_step.py

@@ -12,6 +12,7 @@
 # language governing permissions and limitations under the License.
 from __future__ import absolute_import
 
+import os
 import unittest
 
 from mock import Mock, patch