Merge pull request #29 from NicholasSynovic/main

Bump version

.gitignore

@@ -169,3 +169,5 @@ ptm_torrent/huggingface/v2/data
 ptm_torrent/huggingface/v3/data
 ptm_torrent/pytorchhub/data
 example/onnx
+example/PTMTorrent
+example/onnxmodelzoo.tar.gz

example/README.md

@@ -39,7 +39,7 @@ Python dependencies are managed with `pip` and
 > Instructions were written for Linux operating systems
 
 1. Change directories into the ONNX Model Zoo PTMTorrent directory:
- `cd data/onnxmodelzoo`
+ `cd ptm_torrent/onnxmodelzoo`
 1. Create a `Python 3.10` virtual environment: `python3.10 -m venv env`
 1. Activate virtual environment: `source env/bin/activate`
 1. Upgrade `pip`: `python -m pip install --upgrade pip`

example/example.py

@@ -34,7 +34,7 @@ def extractAuthorRepo(url: str) -> PurePath:
 def main():
  paths: List[PurePath] = []
 
- jsonFile: PurePath = PurePath("onnxmodelzoo.json")
+ jsonFile: PurePath = PurePath("json/onnxmodelzoo.json")
  json: List[dict] = readJSON(filepath=jsonFile)
 
  data: dict

example/run.bash

@@ -1,4 +1,4 @@
 #!/bin/bash
 
 python3.10 example.py | xargs -I % clime-git-commits-extract -d % &&
-clime-git-commits-graph
+clime-git-commits-graph --x-label Days --y-label "Lines of Code" --title "ONNX Model Zoo Lines of Code per Day"

ptm_torrent/huggingface/__main__.py

@@ -3,23 +3,23 @@
 import ptm_torrent.huggingface.createSchema as createSchema
 import ptm_torrent.huggingface.downloadJSON as downloadJSON
 import ptm_torrent.huggingface.downloadRepos as downloadRepos
-import ptm_torrent.huggingface.splitRepos as splitRepos
 import ptm_torrent.huggingface.setupFileSystem as setupFS
+import ptm_torrent.huggingface.splitRepos as splitRepos
 
 if __name__ == "__main__":
  shrinkage: float | str | None
 
  shrinkageEnvironmentVariable: str = "HF_TORRENT_SHRINKAGE"
  shrinkageEnvironmentValue: str = environ.get(shrinkageEnvironmentVariable)
 
- if shrinkageEnvironmentValue is None:
- shrinkage = 0.1
-
  try:
  shrinkage = float(shrinkageEnvironmentValue)
  except ValueError:
  print(f"Invalid value for {shrinkageEnvironmentVariable}. Defaulting to 0.1")
  shrinkage = 0.1
+ except TypeError:
+ print(f"Invalid value for {shrinkageEnvironmentVariable}. Defaulting to 0.1")
+ shrinkage = 0.1
 
  if shrinkage > 1 or shrinkage <= 0:
  print(f"Invalid value for {shrinkageEnvironmentVariable}. Defaulting to 0.1")

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "ptm-torrent"
-version = "0.1.5"
+version = "0.1.6"
 description = ""
 authors = ["Nicholas M. Synovic <nicholas.synovic@gmail.com>"]
 readme = "README.md"