Fix up style and imports

high-performance-spark · Aug 27, 2024 · fe0f192 · fe0f192
1 parent 6e357bb
commit fe0f192
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/python/examples/test_dual_write_new.py b/python/examples/test_dual_write_new.py
@@ -3,6 +3,8 @@
 
 #tag::test[]
 import unittest
+from pyspark.sql import SparkSession
+from pyspark.sql.functions import current_timestamp
 from pyspark.sql.types import Row
 from pyspark.testing.utils import assertDataFrameEqual
 from .dual_write import DualWriteExample

diff --git a/python/tox.ini b/python/tox.ini
@@ -57,7 +57,7 @@ deps =
 [testenv:flake8]
 extras = tests
 skipsdist = True
-commands = flake8 --ignore=F403,E402,F401,F405,W503 examples
+commands = flake8 --ignore=F403,E402,F401,F405,W503,E265 examples
 allowlist_externals = flake8
 
 [testenv:mypy]