From 36107269b18d40b68e8fccaedc624639c8b4cb60 Mon Sep 17 00:00:00 2001
From: Eric Vergnaud <eric.vergnaud@wanadoo.fr>
Date: Tue, 9 Jul 2024 20:00:11 +0200
Subject: [PATCH] whitelist einops (#2060)

## Changes
whitelist einops

### Linked issues
Progresses #1901

### Functionality
None

### Tests
- [x] manually tested

Co-authored-by: Eric Vergnaud <eric.vergnaud@databricks.com>
---
 .../labs/ucx/source_code/known.json           | 20 +++++++++++++++++++
 1 file changed, 20 insertions(+)

diff --git a/src/databricks/labs/ucx/source_code/known.json b/src/databricks/labs/ucx/source_code/known.json
index 5f770b1ee8..644f0bf6d4 100644
--- a/src/databricks/labs/ucx/source_code/known.json
+++ b/src/databricks/labs/ucx/source_code/known.json
@@ -2886,6 +2886,26 @@
   "emmv": {
     "emmv": []
   },
+ "einops": {
+    "einops": [],
+    "einops._backends": [],
+    "einops._torch_specific": [],
+    "einops.array_api": [],
+    "einops.einops": [],
+    "einops.experimental": [],
+    "einops.experimental.indexing": [],
+    "einops.layers": [],
+    "einops.layers._einmix": [],
+    "einops.layers.chainer": [],
+    "einops.layers.flax": [],
+    "einops.layers.keras": [],
+    "einops.layers.oneflow": [],
+    "einops.layers.paddle": [],
+    "einops.layers.tensorflow": [],
+    "einops.layers.torch": [],
+    "einops.packing": [],
+    "einops.parsing": []
+  },
   "entrypoints": {
     "entrypoints": []
   },