Upload hf_utils.py with huggingface_hub
Browse files- hf_utils.py +6 -3
hf_utils.py
CHANGED
|
@@ -30,12 +30,15 @@ class HFCachingContextManager:
|
|
| 30 |
disable_caching()
|
| 31 |
|
| 32 |
|
| 33 |
-
def get_missing_imports(file, exclude=
|
|
|
|
|
|
|
| 34 |
src_dir = Path(__file__).parent
|
| 35 |
python_files = get_all_files_in_dir(src_dir, file_extension=".py")
|
| 36 |
# get only the file without the path and extension
|
| 37 |
required_modules = [Path(p).stem for p in python_files]
|
| 38 |
imports = get_imports(file)
|
| 39 |
imported_modules = [i[1] for i in imports if i[0] == "internal"]
|
| 40 |
-
|
| 41 |
-
|
|
|
|
|
|
| 30 |
disable_caching()
|
| 31 |
|
| 32 |
|
| 33 |
+
def get_missing_imports(file, exclude=None):
|
| 34 |
+
if exclude is None:
|
| 35 |
+
exclude = []
|
| 36 |
src_dir = Path(__file__).parent
|
| 37 |
python_files = get_all_files_in_dir(src_dir, file_extension=".py")
|
| 38 |
# get only the file without the path and extension
|
| 39 |
required_modules = [Path(p).stem for p in python_files]
|
| 40 |
imports = get_imports(file)
|
| 41 |
imported_modules = [i[1] for i in imports if i[0] == "internal"]
|
| 42 |
+
return [
|
| 43 |
+
i for i in required_modules if i not in imported_modules and i not in exclude
|
| 44 |
+
]
|