code quality

mosaicml · Dec 8, 2023 · d172d89 · d172d89
1 parent 0fa1123
commit d172d89
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 2 deletions.
diff --git a/llmfoundry/data/packing.py b/llmfoundry/data/packing.py
@@ -10,7 +10,6 @@
 import torch
 from composer.utils import using_torch_2
 from omegaconf import DictConfig
-from streaming.base.util import clean_stale_shared_memory
 from transformers import PreTrainedTokenizerBase
 
 log = logging.getLogger(__name__)

diff --git a/tests/data/test_packing.py b/tests/data/test_packing.py
@@ -5,7 +5,6 @@
 from typing import Any, Dict, List
 from unittest.mock import Mock, patch
 
-import numpy as np
 import pytest
 import torch
 from composer.utils import dist, reproducibility, using_torch_2