Skip to content

Commit e5afb88

Browse files
authored
tests/conftest.py: automatically add local deepspeed repo when running tests (#7317)
This is a follow up to #923 my original code was a copy from transformers, which has a different fs layout and I missed that. So this PR is fixing it to actually do the right thing. Now you can have multiple clones of deepspeed and the tests will use the local repo automatically and not the pre-installed deepspeed.
1 parent b4cc079 commit e5afb88

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/conftest.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,7 @@
88
import sys
99
import pytest
1010
import os
11-
from os.path import abspath, dirname, join
11+
from os.path import abspath, dirname
1212
import torch
1313
import warnings
1414

@@ -17,7 +17,7 @@
1717

1818
# allow having multiple repository checkouts and not needing to remember to rerun
1919
# 'pip install -e .[dev]' when switching between checkouts and running tests.
20-
git_repo_path = abspath(join(dirname(dirname(__file__)), "src"))
20+
git_repo_path = abspath(dirname(dirname(__file__)))
2121
sys.path.insert(1, git_repo_path)
2222

2323

0 commit comments

Comments
 (0)