From e5e98c3754a72897e78d9b76345359c12738a20b Mon Sep 17 00:00:00 2001
From: Ed Page <eopage@gmail.com>
Date: Thu, 8 Sep 2022 11:17:45 -0500
Subject: [PATCH] fix(config): Don't correct `NDArray` in Python

Fixes #565
---
 src/config.rs | 14 ++++++++++++++
 1 file changed, 14 insertions(+)

diff --git a/src/config.rs b/src/config.rs
index cba60a8..7e0a0da 100644
--- a/src/config.rs
+++ b/src/config.rs
@@ -224,6 +224,20 @@ impl TypeEngineConfig {
                     ..Default::default()
                 },
             });
+        patterns
+            .entry("py".into())
+            .or_insert_with(|| GlobEngineConfig {
+                extend_glob: Vec::new(),
+                engine: EngineConfig {
+                    dict: Some(DictConfig {
+                        extend_identifiers: maplit::hashmap! {
+                            "NDArray".into() => "NDArray".into(),
+                        },
+                        ..Default::default()
+                    }),
+                    ..Default::default()
+                },
+            });
         patterns
             .entry("cert".into())
             .or_insert_with(|| GlobEngineConfig {