gridfm-graphkit 0.0.5__tar.gz → 0.0.6__tar.gz

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (43) hide show
  1. {gridfm_graphkit-0.0.5 → gridfm_graphkit-0.0.6}/PKG-INFO +23 -20
  2. {gridfm_graphkit-0.0.5 → gridfm_graphkit-0.0.6}/README.md +11 -6
  3. {gridfm_graphkit-0.0.5 → gridfm_graphkit-0.0.6}/gridfm_graphkit.egg-info/PKG-INFO +23 -20
  4. gridfm_graphkit-0.0.6/gridfm_graphkit.egg-info/requires.txt +22 -0
  5. {gridfm_graphkit-0.0.5 → gridfm_graphkit-0.0.6}/pyproject.toml +12 -14
  6. gridfm_graphkit-0.0.5/gridfm_graphkit.egg-info/requires.txt +0 -24
  7. {gridfm_graphkit-0.0.5 → gridfm_graphkit-0.0.6}/LICENSE +0 -0
  8. {gridfm_graphkit-0.0.5 → gridfm_graphkit-0.0.6}/gridfm_graphkit/__init__.py +0 -0
  9. {gridfm_graphkit-0.0.5 → gridfm_graphkit-0.0.6}/gridfm_graphkit/__main__.py +0 -0
  10. {gridfm_graphkit-0.0.5 → gridfm_graphkit-0.0.6}/gridfm_graphkit/cli.py +0 -0
  11. {gridfm_graphkit-0.0.5 → gridfm_graphkit-0.0.6}/gridfm_graphkit/datasets/__init__.py +0 -0
  12. {gridfm_graphkit-0.0.5 → gridfm_graphkit-0.0.6}/gridfm_graphkit/datasets/globals.py +0 -0
  13. {gridfm_graphkit-0.0.5 → gridfm_graphkit-0.0.6}/gridfm_graphkit/datasets/normalizers.py +0 -0
  14. {gridfm_graphkit-0.0.5 → gridfm_graphkit-0.0.6}/gridfm_graphkit/datasets/postprocessing.py +0 -0
  15. {gridfm_graphkit-0.0.5 → gridfm_graphkit-0.0.6}/gridfm_graphkit/datasets/powergrid_datamodule.py +0 -0
  16. {gridfm_graphkit-0.0.5 → gridfm_graphkit-0.0.6}/gridfm_graphkit/datasets/powergrid_dataset.py +0 -0
  17. {gridfm_graphkit-0.0.5 → gridfm_graphkit-0.0.6}/gridfm_graphkit/datasets/transforms.py +0 -0
  18. {gridfm_graphkit-0.0.5 → gridfm_graphkit-0.0.6}/gridfm_graphkit/datasets/utils.py +0 -0
  19. {gridfm_graphkit-0.0.5 → gridfm_graphkit-0.0.6}/gridfm_graphkit/io/__init__.py +0 -0
  20. {gridfm_graphkit-0.0.5 → gridfm_graphkit-0.0.6}/gridfm_graphkit/io/param_handler.py +0 -0
  21. {gridfm_graphkit-0.0.5 → gridfm_graphkit-0.0.6}/gridfm_graphkit/io/registries.py +0 -0
  22. {gridfm_graphkit-0.0.5 → gridfm_graphkit-0.0.6}/gridfm_graphkit/models/__init__.py +0 -0
  23. {gridfm_graphkit-0.0.5 → gridfm_graphkit-0.0.6}/gridfm_graphkit/models/gnn_transformer.py +0 -0
  24. {gridfm_graphkit-0.0.5 → gridfm_graphkit-0.0.6}/gridfm_graphkit/models/gps_transformer.py +0 -0
  25. {gridfm_graphkit-0.0.5 → gridfm_graphkit-0.0.6}/gridfm_graphkit/tasks/__init__.py +0 -0
  26. {gridfm_graphkit-0.0.5 → gridfm_graphkit-0.0.6}/gridfm_graphkit/tasks/feature_reconstruction_task.py +0 -0
  27. {gridfm_graphkit-0.0.5 → gridfm_graphkit-0.0.6}/gridfm_graphkit/training/__init__.py +0 -0
  28. {gridfm_graphkit-0.0.5 → gridfm_graphkit-0.0.6}/gridfm_graphkit/training/callbacks.py +0 -0
  29. {gridfm_graphkit-0.0.5 → gridfm_graphkit-0.0.6}/gridfm_graphkit/training/loss.py +0 -0
  30. {gridfm_graphkit-0.0.5 → gridfm_graphkit-0.0.6}/gridfm_graphkit/utils/__init__.py +0 -0
  31. {gridfm_graphkit-0.0.5 → gridfm_graphkit-0.0.6}/gridfm_graphkit/utils/utils.py +0 -0
  32. {gridfm_graphkit-0.0.5 → gridfm_graphkit-0.0.6}/gridfm_graphkit/utils/visualization.py +0 -0
  33. {gridfm_graphkit-0.0.5 → gridfm_graphkit-0.0.6}/gridfm_graphkit.egg-info/SOURCES.txt +0 -0
  34. {gridfm_graphkit-0.0.5 → gridfm_graphkit-0.0.6}/gridfm_graphkit.egg-info/dependency_links.txt +0 -0
  35. {gridfm_graphkit-0.0.5 → gridfm_graphkit-0.0.6}/gridfm_graphkit.egg-info/entry_points.txt +0 -0
  36. {gridfm_graphkit-0.0.5 → gridfm_graphkit-0.0.6}/gridfm_graphkit.egg-info/top_level.txt +0 -0
  37. {gridfm_graphkit-0.0.5 → gridfm_graphkit-0.0.6}/setup.cfg +0 -0
  38. {gridfm_graphkit-0.0.5 → gridfm_graphkit-0.0.6}/tests/test_data_module.py +0 -0
  39. {gridfm_graphkit-0.0.5 → gridfm_graphkit-0.0.6}/tests/test_full_pipeline.py +0 -0
  40. {gridfm_graphkit-0.0.5 → gridfm_graphkit-0.0.6}/tests/test_losses.py +0 -0
  41. {gridfm_graphkit-0.0.5 → gridfm_graphkit-0.0.6}/tests/test_model_outputs.py +0 -0
  42. {gridfm_graphkit-0.0.5 → gridfm_graphkit-0.0.6}/tests/test_normalization.py +0 -0
  43. {gridfm_graphkit-0.0.5 → gridfm_graphkit-0.0.6}/tests/test_yaml_configs.py +0 -0
{gridfm_graphkit-0.0.5 → gridfm_graphkit-0.0.6}/PKG-INFO RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: gridfm-graphkit
3
- Version: 0.0.5
3
+ Version: 0.0.6
4
4
  Summary: Grid Foundation Model
5
5
  Author-email: Matteo Mazzonelli <matteo.mazzonelli1@ibm.com>, Alban Puech <apuech@seas.harvard.edu>, Tamara Govindasamy <tamara.govindasamy@ibm.com>, Mangaliso Mngomezulu <mngomezulum@ibm.com>, Etienne Vos <etienne.vos@ibm.com>, Celia Cintas <celia.cintas@ibm.com>, Jonas Weiss <jwe@zurich.ibm.com>
6
6
  Maintainer-email: Matteo Mazzonelli <matteo.mazzonelli1@ibm.com>
@@ -14,31 +14,39 @@ Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
14
14
  Requires-Python: <3.13,>=3.10
15
15
  Description-Content-Type: text/markdown
16
16
  License-File: LICENSE
17
- Requires-Dist: mlflow>=3.1.0
18
- Requires-Dist: nbformat>=5.10.4
19
- Requires-Dist: networkx>=3.4.2
20
- Requires-Dist: numpy>=2.2.6
21
- Requires-Dist: pandas>=2.3.0
22
- Requires-Dist: plotly>=6.1.2
23
- Requires-Dist: pyyaml>=6.0.2
24
- Requires-Dist: torch>=2.7.1
25
- Requires-Dist: torch-geometric>=2.6.1
26
- Requires-Dist: torchaudio>=2.7.1
27
- Requires-Dist: torchvision>=0.22.1
17
+ Requires-Dist: torch>2.0
18
+ Requires-Dist: torch-geometric
19
+ Requires-Dist: mlflow
20
+ Requires-Dist: nbformat
21
+ Requires-Dist: networkx
22
+ Requires-Dist: numpy
23
+ Requires-Dist: pandas
24
+ Requires-Dist: plotly
25
+ Requires-Dist: pyyaml
28
26
  Requires-Dist: lightning
29
27
  Requires-Dist: seaborn
30
28
  Provides-Extra: dev
31
29
  Requires-Dist: mkdocs-material; extra == "dev"
32
30
  Requires-Dist: mkdocstrings[python]; extra == "dev"
33
- Requires-Dist: pre-commit>=4.2.0; extra == "dev"
34
- Requires-Dist: bandit>=1.8.5; extra == "dev"
31
+ Requires-Dist: pre-commit; extra == "dev"
32
+ Requires-Dist: bandit; extra == "dev"
35
33
  Requires-Dist: build; extra == "dev"
36
34
  Provides-Extra: test
37
35
  Requires-Dist: pytest; extra == "test"
38
36
  Requires-Dist: pytest-cov; extra == "test"
39
37
  Dynamic: license-file
40
38
 
41
- # gridfm-graphkit
39
+ <p align="center">
40
+ <img src="https://raw.githubusercontent.com/gridfm/gridfm-graphkit/refs/heads/main/docs/figs/KIT.png" alt="GridFM logo" style="width: 40%; height: auto;"/>
41
+ <br/>
42
+ </p>
43
+
44
+ <p align="center" style="font-size: 25px;">
45
+ <b>gridfm-graphkit</b>
46
+ </p>
47
+
48
+
49
+ [![DOI](https://zenodo.org/badge/1007159095.svg)](https://doi.org/10.5281/zenodo.17016737)
42
50
  [![Docs](https://img.shields.io/badge/docs-available-brightgreen)](https://gridfm.github.io/gridfm-graphkit/)
43
51
  ![Coverage](https://img.shields.io/badge/coverage-83%25-yellowgreen)
44
52
  ![Python](https://img.shields.io/badge/python-3.10%20%E2%80%93%203.12-blue)
@@ -48,11 +56,6 @@ This library is brought to you by the GridFM team to train, finetune and interac
48
56
 
49
57
  ---
50
58
 
51
- <p align="center">
52
- <img src="https://raw.githubusercontent.com/gridfm/gridfm-graphkit/refs/heads/main/docs/figs/pre_training.png" alt="GridFM logo"/>
53
- <br/>
54
- </p>
55
-
56
59
  # Installation
57
60
 
58
61
  You can install `gridfm-graphkit` directly from PyPI:
{gridfm_graphkit-0.0.5 → gridfm_graphkit-0.0.6}/README.md RENAMED
@@ -1,4 +1,14 @@
1
- # gridfm-graphkit
1
+ <p align="center">
2
+ <img src="https://raw.githubusercontent.com/gridfm/gridfm-graphkit/refs/heads/main/docs/figs/KIT.png" alt="GridFM logo" style="width: 40%; height: auto;"/>
3
+ <br/>
4
+ </p>
5
+
6
+ <p align="center" style="font-size: 25px;">
7
+ <b>gridfm-graphkit</b>
8
+ </p>
9
+
10
+
11
+ [![DOI](https://zenodo.org/badge/1007159095.svg)](https://doi.org/10.5281/zenodo.17016737)
2
12
  [![Docs](https://img.shields.io/badge/docs-available-brightgreen)](https://gridfm.github.io/gridfm-graphkit/)
3
13
  ![Coverage](https://img.shields.io/badge/coverage-83%25-yellowgreen)
4
14
  ![Python](https://img.shields.io/badge/python-3.10%20%E2%80%93%203.12-blue)
@@ -8,11 +18,6 @@ This library is brought to you by the GridFM team to train, finetune and interac
8
18
 
9
19
  ---
10
20
 
11
- <p align="center">
12
- <img src="https://raw.githubusercontent.com/gridfm/gridfm-graphkit/refs/heads/main/docs/figs/pre_training.png" alt="GridFM logo"/>
13
- <br/>
14
- </p>
15
-
16
21
  # Installation
17
22
 
18
23
  You can install `gridfm-graphkit` directly from PyPI:
{gridfm_graphkit-0.0.5 → gridfm_graphkit-0.0.6}/gridfm_graphkit.egg-info/PKG-INFO RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: gridfm-graphkit
3
- Version: 0.0.5
3
+ Version: 0.0.6
4
4
  Summary: Grid Foundation Model
5
5
  Author-email: Matteo Mazzonelli <matteo.mazzonelli1@ibm.com>, Alban Puech <apuech@seas.harvard.edu>, Tamara Govindasamy <tamara.govindasamy@ibm.com>, Mangaliso Mngomezulu <mngomezulum@ibm.com>, Etienne Vos <etienne.vos@ibm.com>, Celia Cintas <celia.cintas@ibm.com>, Jonas Weiss <jwe@zurich.ibm.com>
6
6
  Maintainer-email: Matteo Mazzonelli <matteo.mazzonelli1@ibm.com>
@@ -14,31 +14,39 @@ Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
14
14
  Requires-Python: <3.13,>=3.10
15
15
  Description-Content-Type: text/markdown
16
16
  License-File: LICENSE
17
- Requires-Dist: mlflow>=3.1.0
18
- Requires-Dist: nbformat>=5.10.4
19
- Requires-Dist: networkx>=3.4.2
20
- Requires-Dist: numpy>=2.2.6
21
- Requires-Dist: pandas>=2.3.0
22
- Requires-Dist: plotly>=6.1.2
23
- Requires-Dist: pyyaml>=6.0.2
24
- Requires-Dist: torch>=2.7.1
25
- Requires-Dist: torch-geometric>=2.6.1
26
- Requires-Dist: torchaudio>=2.7.1
27
- Requires-Dist: torchvision>=0.22.1
17
+ Requires-Dist: torch>2.0
18
+ Requires-Dist: torch-geometric
19
+ Requires-Dist: mlflow
20
+ Requires-Dist: nbformat
21
+ Requires-Dist: networkx
22
+ Requires-Dist: numpy
23
+ Requires-Dist: pandas
24
+ Requires-Dist: plotly
25
+ Requires-Dist: pyyaml
28
26
  Requires-Dist: lightning
29
27
  Requires-Dist: seaborn
30
28
  Provides-Extra: dev
31
29
  Requires-Dist: mkdocs-material; extra == "dev"
32
30
  Requires-Dist: mkdocstrings[python]; extra == "dev"
33
- Requires-Dist: pre-commit>=4.2.0; extra == "dev"
34
- Requires-Dist: bandit>=1.8.5; extra == "dev"
31
+ Requires-Dist: pre-commit; extra == "dev"
32
+ Requires-Dist: bandit; extra == "dev"
35
33
  Requires-Dist: build; extra == "dev"
36
34
  Provides-Extra: test
37
35
  Requires-Dist: pytest; extra == "test"
38
36
  Requires-Dist: pytest-cov; extra == "test"
39
37
  Dynamic: license-file
40
38
 
41
- # gridfm-graphkit
39
+ <p align="center">
40
+ <img src="https://raw.githubusercontent.com/gridfm/gridfm-graphkit/refs/heads/main/docs/figs/KIT.png" alt="GridFM logo" style="width: 40%; height: auto;"/>
41
+ <br/>
42
+ </p>
43
+
44
+ <p align="center" style="font-size: 25px;">
45
+ <b>gridfm-graphkit</b>
46
+ </p>
47
+
48
+
49
+ [![DOI](https://zenodo.org/badge/1007159095.svg)](https://doi.org/10.5281/zenodo.17016737)
42
50
  [![Docs](https://img.shields.io/badge/docs-available-brightgreen)](https://gridfm.github.io/gridfm-graphkit/)
43
51
  ![Coverage](https://img.shields.io/badge/coverage-83%25-yellowgreen)
44
52
  ![Python](https://img.shields.io/badge/python-3.10%20%E2%80%93%203.12-blue)
@@ -48,11 +56,6 @@ This library is brought to you by the GridFM team to train, finetune and interac
48
56
 
49
57
  ---
50
58
 
51
- <p align="center">
52
- <img src="https://raw.githubusercontent.com/gridfm/gridfm-graphkit/refs/heads/main/docs/figs/pre_training.png" alt="GridFM logo"/>
53
- <br/>
54
- </p>
55
-
56
59
  # Installation
57
60
 
58
61
  You can install `gridfm-graphkit` directly from PyPI:
gridfm_graphkit-0.0.6/gridfm_graphkit.egg-info/requires.txt ADDED
@@ -0,0 +1,22 @@
1
+ torch>2.0
2
+ torch-geometric
3
+ mlflow
4
+ nbformat
5
+ networkx
6
+ numpy
7
+ pandas
8
+ plotly
9
+ pyyaml
10
+ lightning
11
+ seaborn
12
+
13
+ [dev]
14
+ mkdocs-material
15
+ mkdocstrings[python]
16
+ pre-commit
17
+ bandit
18
+ build
19
+
20
+ [test]
21
+ pytest
22
+ pytest-cov
{gridfm_graphkit-0.0.5 → gridfm_graphkit-0.0.6}/pyproject.toml RENAMED
@@ -9,7 +9,7 @@ namespaces = false
9
9
  [project]
10
10
  name = "gridfm-graphkit"
11
11
  description = "Grid Foundation Model"
12
- version = "0.0.5"
12
+ version = "0.0.6"
13
13
  readme = "README.md"
14
14
  license = "Apache-2.0"
15
15
  requires-python = ">=3.10,<3.13"
@@ -40,17 +40,15 @@ classifiers = [
40
40
 
41
41
 
42
42
  dependencies = [
43
- "mlflow>=3.1.0",
44
- "nbformat>=5.10.4",
45
- "networkx>=3.4.2",
46
- "numpy>=2.2.6",
47
- "pandas>=2.3.0",
48
- "plotly>=6.1.2",
49
- "pyyaml>=6.0.2",
50
- "torch>=2.7.1",
51
- "torch-geometric>=2.6.1",
52
- "torchaudio>=2.7.1",
53
- "torchvision>=0.22.1",
43
+ "torch>2.0",
44
+ "torch-geometric",
45
+ "mlflow",
46
+ "nbformat",
47
+ "networkx",
48
+ "numpy",
49
+ "pandas",
50
+ "plotly",
51
+ "pyyaml",
54
52
  "lightning",
55
53
  "seaborn",
56
54
  ]
@@ -59,8 +57,8 @@ dependencies = [
59
57
  dev = [
60
58
  "mkdocs-material",
61
59
  "mkdocstrings[python]",
62
- "pre-commit>=4.2.0",
63
- "bandit>=1.8.5",
60
+ "pre-commit",
61
+ "bandit",
64
62
  "build"
65
63
  ]
66
64
 
gridfm_graphkit-0.0.5/gridfm_graphkit.egg-info/requires.txt DELETED
@@ -1,24 +0,0 @@
1
- mlflow>=3.1.0
2
- nbformat>=5.10.4
3
- networkx>=3.4.2
4
- numpy>=2.2.6
5
- pandas>=2.3.0
6
- plotly>=6.1.2
7
- pyyaml>=6.0.2
8
- torch>=2.7.1
9
- torch-geometric>=2.6.1
10
- torchaudio>=2.7.1
11
- torchvision>=0.22.1
12
- lightning
13
- seaborn
14
-
15
- [dev]
16
- mkdocs-material
17
- mkdocstrings[python]
18
- pre-commit>=4.2.0
19
- bandit>=1.8.5
20
- build
21
-
22
- [test]
23
- pytest
24
- pytest-cov
{gridfm_graphkit-0.0.5 → gridfm_graphkit-0.0.6}/LICENSE RENAMED
File without changes