From ceb5e85c0a98d9154b3dc0083eba39abd6160df1 Mon Sep 17 00:00:00 2001
From: Rishabh Ranjan <rishabhranjan0207@gmail.com>
Date: Fri, 5 Jul 2024 13:29:56 -0700
Subject: [PATCH] Cleanup and update packaging (#226)

---
 .github/workflows/full_testing.yml |  6 +++---
 .github/workflows/testing.yml      |  2 +-
 pyproject.toml                     | 17 +++++++----------
 3 files changed, 11 insertions(+), 14 deletions(-)

diff --git a/.github/workflows/full_testing.yml b/.github/workflows/full_testing.yml
index 0ab6bddb..c422d5ba 100644
--- a/.github/workflows/full_testing.yml
+++ b/.github/workflows/full_testing.yml
@@ -27,12 +27,12 @@ jobs:
 
       - name: Install PyTorch
         run: |
-          pip install torch==2.1.0 --extra-index-url https://download.pytorch.org/whl/cpu
-          pip install --no-index pyg-lib==0.3.1 -f https://data.pyg.org/whl/torch-2.1.0+cpu.html
+          pip install torch==2.3.0 --extra-index-url https://download.pytorch.org/whl/cpu
+          pip install --no-index pyg-lib==0.4.0 -f https://data.pyg.org/whl/torch-2.3.0+cpu.html
 
       - name: Install main package
         run: |
-          pip install .[example,test]
+          pip install .[full,test]
 
       - name: Run tests
         run: |
diff --git a/.github/workflows/testing.yml b/.github/workflows/testing.yml
index d1296a39..6c04486f 100644
--- a/.github/workflows/testing.yml
+++ b/.github/workflows/testing.yml
@@ -28,7 +28,7 @@ jobs:
 
       - name: Install main package
         run: |
-          pip install .[test,example]
+          pip install .[test]
 
       - name: Run tests
         run: |
diff --git a/pyproject.toml b/pyproject.toml
index 4c681457..8a54fb97 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -20,24 +20,21 @@ dependencies=[
 	"pyarrow",
 	"numpy",
 	"duckdb",
-	"requests",
-	"tqdm",
 	"scikit-learn",
 	"typing-extensions",
-	"matplotlib",
-	"networkx",
 ]
 
 [project.optional-dependencies]
-example=[
+full=[
 	"torch",
-	"pytorch_frame @ git+https://github.com/pyg-team/pytorch-frame", # TODO: release to pypi
+	# We rely on https://github.com/pyg-team/pytorch-frame/pull/395 for rel-amazon.
+	# TODO: use the PyPI version when it's released.
+	"pytorch_frame[full] @ git+https://github.com/pyg-team/pytorch-frame",
 	"torch_geometric",
-	"faiss-cpu",
+]
+example=[
 	"sentence-transformers",
-	# TODO: should be dependencies of torch_frame actually
-	"lightgbm",
-	"optuna",
+	"tqdm",
 ]
 test=[
 	"pytest",