fix: finish pattern

Bearer · Jun 4, 2024 · 428f1a7 · 428f1a7
1 parent 96fb2db
commit 428f1a7
Showing 1 changed file with 10 additions and 1 deletion.
diff --git 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