Tsfm revision

pyproject.toml

@@ -33,6 +33,8 @@ dependencies = [
     "scipy>=1.10.0",
     "sktime>=1.0.1",
     "granite-tsfm>=0.3.5",
+    "statsmodels",
+    "pywavelets",
 ]
 
 [project.scripts]

src/couchdb/.allowed_datafiles

@@ -13,4 +13,6 @@ src/couchdb/scenarios_data/shared/iot/hydraulic_pump_1.json
 src/couchdb/scenarios_data/shared/iot/metro_pump_1.json
 src/couchdb/scenarios_data/shared/iot/motor_01.json
 src/couchdb/scenarios_data/shared/work_order/workorders.csv
+src/couchdb/scenarios_data/shared/tsfm/feature_catalog.json
+src/couchdb/scenarios_data/shared/tsfm/model_catalog.json
 src/couchdb/scenarios_data/shared/iot/asset_profile_sample.json

src/couchdb/collections.json

@@ -143,5 +143,44 @@
         ],
         "id_prefix": "fc",
         "indexes": []
+    },
+    "model_catalog": {
+        "format": "json",
+        "primary_key": [
+            "model_id"
+        ],
+        "id_prefix": "model",
+        "indexes": [
+            [
+                "task_ids"
+            ],
+            [
+                "status"
+            ],
+            [
+                "model_family"
+            ],
+            [
+                "domain"
+            ]
+        ]
+    },
+    "feature_catalog": {
+        "format": "json",
+        "primary_key": [
+            "feature_id"
+        ],
+        "id_prefix": "feature",
+        "indexes": [
+            [
+                "scenario_categories"
+            ],
+            [
+                "kind"
+            ],
+            [
+                "status"
+            ]
+        ]
     }
 }

src/couchdb/scenarios_data/default/manifest.json

@@ -7,6 +7,8 @@
     ],
     "asset": "shared/iot/asset_profile_sample.json",
     "vibration": "shared/iot/motor_01.json",
+    "model_catalog": "shared/tsfm/model_catalog.json",
+    "feature_catalog": "shared/tsfm/feature_catalog.json",
     "catalog": [
         "shared/catalog/assets.csv",
         "shared/catalog/failure_modes.csv",