Update dependency to fix unit test issue

requirements.txt

@@ -24,6 +24,7 @@ Markdown==3.5.2
 MarkupSafe==2.1.5
 ml-dtypes==0.2.0
 mpmath==1.3.0
+munkres==1.1.4
 networkx==3.2.1
 numpy==1.26.4
 oauthlib==3.2.2
@@ -39,6 +40,7 @@ pyasn1-modules==0.3.0
 pybind11==2.11.1
 pytest==8.0.0
 pytest-asyncio==0.23.5
+pytest-mock==3.12.0
 python-dateutil==2.8.2
 pytz==2024.1
 PyYAML==6.0.1
@@ -48,8 +50,10 @@ requests-oauthlib==1.3.1
 river==0.21.0
 rsa==4.9
 safetensors==0.4.2
+scikit-learn==1.4.1.post1
 scipy==1.12.0
 six==1.16.0
+snorkel==0.9.9
 sympy==1.12
 tensorboard==2.15.2
 tensorboard-data-server==0.7.2
@@ -59,7 +63,9 @@ tensorflow-hub==0.16.1
 tensorflow-io-gcs-filesystem==0.36.0
 termcolor==2.4.0
 tf-keras==2.15.0
+threadpoolctl==3.3.0
 tokenizers==0.15.2
+torch==2.2.0
 tqdm==4.66.2
 typing_extensions==4.9.0
 tzdata==2024.1