From 5f8c31af2e5a58999b949af1f2ba6394afad3da2 Mon Sep 17 00:00:00 2001
From: czzhangheng <cz189@qq.com>
Date: Wed, 23 Apr 2025 23:24:43 +0800
Subject: [PATCH] =?UTF-8?q?=E6=B7=BB=E5=8A=A0STIDGCN?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 model/model_selector.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/model/model_selector.py b/model/model_selector.py
index 0117619..091a465 100644
--- a/model/model_selector.py
+++ b/model/model_selector.py
@@ -14,6 +14,7 @@ from model.STSGCN.STSGCN import STSGCN
 from model.STGODE.STGODE import ODEGCN
 from model.PDG2SEQ.PDG2Seq import PDG2Seq
 from model.STMLP.STMLP import STMLP
+from model.STIDGCN.STIDGCN import STIDGCN
 
 def model_selector(model):
     match model['type']:
@@ -33,4 +34,5 @@ def model_selector(model):
         case 'STGODE': return ODEGCN(model)
         case 'PDG2SEQ': return PDG2Seq(model)
         case 'STMLP': return STMLP(model)
+        case 'STIDGCN': return STIDGCN(model)