Merge pull request #135 from yoshitomo-matsubara/dev

Update setup
yoshitomo-matsubara · Aug 14, 2023 · 0c45ce1 · 0c45ce1
2 parents c3e2a63 + 3c00648
commit 0c45ce1
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/setup.py b/setup.py
@@ -3,7 +3,7 @@
 with open('README.md', 'r') as f:
     long_description = f.read()
 
-description = 'SC2: Supervised Compression for Split Computing.'
+description = 'SC2 Benchmark: Supervised Compression for Split Computing.'
 setup(
     name='sc2bench',
     version='0.0.3',
@@ -27,6 +27,7 @@
         'timm>=0.4.12'
     ],
     extras_require={
-        'test': ['pytest']
+        'test': ['pytest'],
+        'docs': ['sphinx', 'sphinx_rtd_theme']
     }
 )