From 428f1a79ddae9f2eacee5ac9aba3b49952cfdc21 Mon Sep 17 00:00:00 2001
From: elsapet <elizabeth@bearer.sh>
Date: Tue, 4 Jun 2024 11:17:50 +0200
Subject: [PATCH] fix: finish pattern

---
 rules/python/third_parties/google_dataflow.yml | 11 ++++++++++-
 1 file changed, 10 insertions(+), 1 deletion(-)

diff --git a/rules/python/third_parties/google_dataflow.yml b/rules/python/third_parties/google_dataflow.yml
index c02b6579..be906dc3 100644
--- a/rules/python/third_parties/google_dataflow.yml
+++ b/rules/python/third_parties/google_dataflow.yml
@@ -1,8 +1,17 @@
 imports:
   - python_shared_lang_datatype
+  - python_shared_lang_import1
 patterns:
-  - pattern: beam.Create($<...>$<DATA_TYPE>$<...>)
+  - pattern: $<CREATE>($<...>$<DATA_TYPE>$<...>)
     filters:
+      - variable: CREATE
+        detection: python_shared_lang_import1
+        scope: cursor
+        filters:
+          - variable: MODULE1
+            values: [apache_beam]
+          - variable: NAME
+            values: [Create]
       - variable: DATA_TYPE
         detection: python_shared_lang_datatype
         scope: result