From fe0f1926e78b975b26564ac153dee68b4cb3418f Mon Sep 17 00:00:00 2001
From: Holden Karau <holden@pigscanfly.ca>
Date: Tue, 27 Aug 2024 09:53:32 -0700
Subject: [PATCH] Fix up style and imports

---
 python/examples/test_dual_write_new.py | 2 ++
 python/tox.ini                         | 2 +-
 2 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/python/examples/test_dual_write_new.py b/python/examples/test_dual_write_new.py
index 751ae32..12d00a6 100644
--- 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
index 330cd58..2aa2d4d 100644
--- 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]