Skip to content

Commit

Permalink
Add pytorch_geometric in setup.py file
Browse files Browse the repository at this point in the history
  • Loading branch information
FilippoOlivo committed Nov 25, 2024
1 parent 99ad55f commit ed1ea7c
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -15,7 +15,8 @@
KEYWORDS = 'physics-informed neural-network'

REQUIRED = [
'numpy', 'matplotlib', 'torch', 'lightning', 'torch_geometric', 'torch-cluster'
'numpy', 'matplotlib', 'torch', 'lightning', 'pytorch_lightning',
'torch_geometric', 'torch-cluster'
]

EXTRAS = {
Expand Down

0 comments on commit ed1ea7c

Please sign in to comment.