update to see versions

huggingface · May 1, 2024 · db90351 · db90351
1 parent 1d9404b
commit db90351
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/.circleci/create_circleci_config.py b/.circleci/create_circleci_config.py
@@ -104,7 +104,9 @@ def to_dict(self):
             {"attach_workspace": {"at": "test_preparation"}},
         ]
         steps.extend([{"run": l} for l in self.install_steps])
-        steps.append({"run": {"name": "Show installed libraries and their versions", "command": """du -h -d 1 "$(pip -V | cut -d ' ' -f 4 | sed 's/pip//g')" | grep -vE "dist-info|_distutils_hack|__pycache__" | sort -h | tee installed.txt || true"""}})
+        steps.append({"run": {"name": "Show installed libraries and their size", "command": """du -h -d 1 "$(pip -V | cut -d ' ' -f 4 | sed 's/pip//g')" | grep -vE "dist-info|_distutils_hack|__pycache__" | sort -h | tee installed.txt || true"""}})
+        steps.append({"run": {"name": "Show installed libraries and their versions", "command": """pip list --format=freeze | awk -F '==' '{print $1" ==  "$2}' | tee installed.txt || true"""}})
+
         steps.append({"run":{"name":"Show biggest libraries","command":"""dpkg-query --show --showformat='${Installed-Size}\t${Package}\n' | sort -rh | head -25 | sort -h | awk '{ printf("%.5f GB %s\\n",$1/1024/1024, $2)}' || true"""}})
         steps.append({"store_artifacts": {"path": "installed.txt"}})