refactor the add nodes methods

ComputationalPsychiatry · Nov 20, 2024 · 16f6eb0 · 16f6eb0
1 parent c4ab757
commit 16f6eb0
Show file tree

Hide file tree

Showing 14 changed files with 604 additions and 595 deletions.
diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml
@@ -9,7 +9,7 @@ repos:
     hooks:
     - id: isort
 -   repo: https://github.com/ambv/black
-    rev: 24.8.0
+    rev: 24.10.0
     hooks:
     - id: black
       language_version: python3
@@ -24,7 +24,7 @@ repos:
         args: ['--ignore', 'D213,D100,D203,D104']
         files: ^src/
 -   repo: https://github.com/pre-commit/mirrors-mypy
-    rev: 'v1.11.2'
+    rev: 'v1.13.0'
     hooks:
     - id: mypy 
       files: ^src/

diff --git a/docs/source/conf.py b/docs/source/conf.py
@@ -17,15 +17,15 @@
 # -- Project information -----------------------------------------------------
 
 project = "pyhgf"
-copyright = u"2022-{}, Nicolas Legrand".format(time.strftime("%Y"))
+copyright = "2022-{}, Nicolas Legrand".format(time.strftime("%Y"))
 author = "Nicolas Legrand"
 release = pyhgf.__version__
 
 nb_execution_timeout = 300
 
 image_scrapers = ("matplotlib",)
 
-bibtex_bibfiles = ['refs.bib']
+bibtex_bibfiles = ["refs.bib"]
 bibtex_reference_style = "author_year"
 bibtex_default_style = "unsrt"
 
@@ -73,10 +73,10 @@
 plot_html_show_source_link = False
 
 source_suffix = {
-    '.rst': 'restructuredtext',
-    '.md': 'myst-nb',
-    '.ipynb': 'myst-nb',
-    '.myst': 'myst-nb',
+    ".rst": "restructuredtext",
+    ".md": "myst-nb",
+    ".ipynb": "myst-nb",
+    ".myst": "myst-nb",
 }
 
 # The master toctree document.
@@ -112,13 +112,13 @@
     "logo": {
         "text": "pyhgf",
     },
-    "show_nav_level": 2
+    "show_nav_level": 2,
 }
 
 myst_enable_extensions = ["dollarmath", "colon_fence"]
 
 html_sidebars = {
-  "api": [],
-  "cite": [],
-  "references": [],
+    "api": [],
+    "cite": [],
+    "references": [],
 }
diff --git a/docs/source/notebooks/0.3-Generalised_filtering.ipynb b/docs/source/notebooks/0.3-Generalised_filtering.ipynb
diff --git a/pyhgf/model/__init__.py b/pyhgf/model/__init__.py
@@ -1,5 +1,27 @@
+from .add_nodes import (
+    add_binary_state,
+    add_categorical_state,
+    add_continuous_state,
+    add_dp_state,
+    add_ef_state,
+    get_couplings,
+    insert_nodes,
+    update_parameters,
+)
 from .network import Network
 
 from .hgf import HGF  # isort: skip
 
-__all__ = ["HGF", "Network"]
+__all__ = [
+    "HGF",
+    "Network",
+    "add_nodes",
+    "add_continuous_state",
+    "add_binary_state",
+    "add_ef_state",
+    "add_categorical_state",
+    "add_dp_state",
+    "get_couplings",
+    "update_parameters",
+    "insert_nodes",
+]