Remove unused import

pthombre · pthombre · commit 33ea6fc1d1fd · 2026-01-06T14:28:15.000-08:00
Signed-off-by: Pranav Prashant Thombre &lt;pthombre@nvidia.com&gt;
diff --git a/dfm/src/automodel/flow_matching/flow_matching_pipeline.py b/dfm/src/automodel/flow_matching/flow_matching_pipeline.py
@@ -31,7 +31,6 @@
 import math
 import os
 import random
-from dataclasses import dataclass
 from typing import Any, Dict, Optional, Tuple
 
 import torch
