Move testing dependencies to optional dependency

.github/workflows/ufmt.yml

@@ -20,7 +20,7 @@ jobs:
         python-version: ${{ matrix.python-version }}
     - name: Install dependencies
       run: |
-        pip install ufmt
+        pip install black==23.3.0 usort==1.0.6 ufmt==2.1.0 libcst==1.0.1
     - name: Analyzing the code with ufmt
       run: |
         ufmt check .

README.md

@@ -12,6 +12,9 @@ will change rapidly.
 pip install .
 # Optionally install editable
 pip install -e .
+
+# Optionally Install dev tooling
+pip install -e ".[dev]"
 ```
 
 # User API, subject to change

float8_experimental/float8_linear_utils.py

@@ -111,7 +111,6 @@ def sync_float8_amax_and_scale_history(
     # Lazy import to avoid circular dependency
 
     if fp8_classes is None:
-
         fp8_classes = Float8Linear
 
     for name, child in model.named_modules():

pyproject.toml

@@ -16,19 +16,23 @@ classifiers = [
 ]
 
 dependencies = [
-    "torch >= 2.0.1",
-    "transformers==4.32.0",
-    "fire==0.5.0",
-    "fairscale==0.4.13",
-    "tqdm==4.66.1",
-    "pandas >= 2.0",
+    "torch >= 2.1",
+    "fairscale==0.4.13"
 ]
 
 [project.optional-dependencies]
+test = [
+    "transformers==4.32.0",
+    "pandas >= 2.0",
+    "tqdm==4.66.1",
+    "fire==0.5.0"
+]
 dev = [
-    "black",
-    "usort",
-    "libcst",
+    "black==23.3.0",
+    "usort==1.0.6",
+    "ufmt==2.1.0",
+    "libcst==1.0.1",
+    "pytest==7.4.0",
     "bumpver",
     "pip-tools"
 ]