Shorten names of files causing installation errors (#1540) (#1545)

IDAES · Nov 27, 2024 · febf2c3 · febf2c3
1 parent 4bf597e
commit febf2c3
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 2 deletions.
diff --git a/..._tanh_1e-06_4096_tr_15481_Calcite_ST.onnx → ...ests/data/onnx_models/net_Calcite_ST.onnx b/..._tanh_1e-06_4096_tr_15481_Calcite_ST.onnx → ...ests/data/onnx_models/net_Calcite_ST.onnx
diff --git a/..._4096_tr_15481_Calcite_ST_idaes_info.json → ...nnx_models/net_Calcite_ST_idaes_info.json b/..._4096_tr_15481_Calcite_ST_idaes_info.json → ...nnx_models/net_Calcite_ST_idaes_info.json
diff --git a/idaes/core/surrogate/tests/test_onnx_surrogate.py b/idaes/core/surrogate/tests/test_onnx_surrogate.py
@@ -40,7 +40,7 @@
 
 
 def load_onnx_model_data(
-    name="net_st_net_5000_STM_100_s_2000000_60_5_tanh_1e-06_4096_tr_15481_Calcite_ST",
+    name="net_Calcite_ST",
 ):
     onnx_folder_name = os.path.join(this_file_dir(), "data", "onnx_models")
     onnx_model = onnx.load(os.path.join(onnx_folder_name, "{}.onnx".format(name)))
@@ -138,7 +138,7 @@ def test_onnx_surrogate_load_and_save_from_file():
 
     onnx_surrogate = ONNXSurrogate.load_onnx_model(
         onnx_model_location=os.path.join(this_file_dir(), "data", "onnx_models"),
-        model_name="net_st_net_5000_STM_100_s_2000000_60_5_tanh_1e-06_4096_tr_15481_Calcite_ST",
+        model_name="net_Calcite_ST",
     )
     with TempfileManager.new_context() as tf:
         dname = tf.mkdtemp()