From 76752f1fbf6dd08b10f0616fad446516f1835156 Mon Sep 17 00:00:00 2001
From: Chenyang Yuan <yuanchenyang@gmail.com>
Date: Wed, 10 Jan 2024 22:20:07 -0500
Subject: [PATCH] Added CI

---
 .github/workflows/CI.yml | 26 ++++++++++++++++++++++++++
 README.md                |  4 ++++
 2 files changed, 30 insertions(+)
 create mode 100644 .github/workflows/CI.yml

diff --git a/.github/workflows/CI.yml b/.github/workflows/CI.yml
new file mode 100644
index 0000000..b4222cd
--- /dev/null
+++ b/.github/workflows/CI.yml
@@ -0,0 +1,26 @@
+name: CI
+
+on: [push, pull_request]
+
+jobs:
+  build:
+    runs-on: ubuntu-20.04
+    strategy:
+      matrix:
+        python-version: ["3.10", "3.11",]
+
+    steps:
+      - uses: actions/checkout@v3
+      - name: Set up Python ${{ matrix.python-version }}
+        uses: actions/setup-python@v4
+        with:
+          python-version: ${{ matrix.python-version }}
+      - name: Install pip
+        run: |
+          python -m pip install --upgrade pip
+      - name: Install package and test dependencies
+        run: |
+          pip install -e .[test]
+      - name: Test with pytest
+        run: |
+          python -m pytest
diff --git a/README.md b/README.md
index 29b7904..ff390ea 100644
--- a/README.md
+++ b/README.md
@@ -85,6 +85,10 @@ of data, without labels. To remove labels from existing datasets, extract the
 data with the provided `MappedDataset` wrapper before constructing a
 `DataLoader`.
 
+Two toy datasets, `Swissroll` and
+[`DatasaurusDozen`](https://www.research.autodesk.com/publications/same-stats-different-graphs/),
+are provided.
+
 ### Model
 All model objects should be a subclass of `torch.nn.Module`. Models should have:
   - A parameter `input_dims`, a tuple containing the dimensions of the input to