Add .keras suffix to package_data

IDAES · Aug 16, 2024 · 0c2775e · 0c2775e
1 parent 1113057
commit 0c2775e
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/setup.py b/setup.py
@@ -176,6 +176,7 @@ def __getitem__(self, key):
             "*.trc",
             "*.xlsx",  # idaes/dmf/tests/data_files - tabular import test files
             "*.nl",
+            "*.keras",  # idaes/core/surrogate/tests/data/keras_models
         ]
     },
     include_package_data=True,