gemini-code-assist[bot] commented on code in PR #38512:
URL: https://github.com/apache/beam/pull/38512#discussion_r3251017243


##########
sdks/python/apache_beam/yaml/yaml_io_test.py:
##########
@@ -18,6 +18,8 @@
 import io
 import json
 import logging
+import os
+import tempfile
 import unittest

Review Comment:
   ![medium](https://www.gstatic.com/codereviewagent/medium-priority.svg)
   
   The `mock` module is used in the new tests (e.g., line 643) but it is not 
imported in this file. This will cause a `NameError` when running the tests. 
Please add the import.
   
   ```suggestion
   import unittest
   from unittest import mock
   ```



-- 
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.

To unsubscribe, e-mail: [email protected]

For queries about this service, please contact Infrastructure at:
[email protected]

Reply via email to