Fix _parse_repo_info (onnx#4648)

* Fix enforce-style CI error: solve empty-body issue raised by mypy 0.990 (onnx#4643)

* solve empty-body issue raised by mypy 0.990

Signed-off-by: Chun-Wei Chen <[email protected]>

* fix black

Signed-off-by: Chun-Wei Chen <[email protected]>

Signed-off-by: Chun-Wei Chen <[email protected]>
Signed-off-by: Yi Zhang <[email protected]>

* Add 32 and 64 bit unsigned integers as tensor types in IR (onnx#4634)

These types are already extensively used throughout in the operator
definitions, but were missing in the definition of the intermediate
representation.

Signed-off-by: Christian Bourjau <[email protected]>

Signed-off-by: Christian Bourjau <[email protected]>
Co-authored-by: G. Ramalingam <[email protected]>
Signed-off-by: Yi Zhang <[email protected]>

* Reference implementation for onnx operators (python runtime for onnx) (onnx#4483)

* add backend test

Signed-off-by: xadupre <[email protected]>

* fix backend

Signed-off-by: xadupre <[email protected]>

* lint

Signed-off-by: xadupre <[email protected]>

* add more unary operators

Signed-off-by: xadupre <[email protected]>

* lint

Signed-off-by: xadupre <[email protected]>

* black

Signed-off-by: xadupre <[email protected]>

* black

Signed-off-by: xadupre <[email protected]>

* black

Signed-off-by: xadupre <[email protected]>

* fix type issue in operator Constant

Signed-off-by: xadupre <[email protected]>

* fix misspelling

Signed-off-by: xadupre <[email protected]>

* switch to relative import

Signed-off-by: sdpython <[email protected]>

* apply isort

Signed-off-by: sdpython <[email protected]>

* pylint

Signed-off-by: sdpython <[email protected]>

* black

Signed-off-by: sdpython <[email protected]>

* less lint issues

Signed-off-by: sdpython <[email protected]>

* make OpRun an abstractclass

Signed-off-by: sdpython <[email protected]>

* add and, or, xor, argmax, argmin

Signed-off-by: sdpython <[email protected]>

* add GlobalAveragePool

Signed-off-by: sdpython <[email protected]>

* add BatchNormalization

Signed-off-by: sdpython <[email protected]>

* add BitShift

Signed-off-by: sdpython <[email protected]>

* add operator cast

Signed-off-by: sdpython <[email protected]>

* add operator CastLike

Signed-off-by: sdpython <[email protected]>

* add operator Celu

Signed-off-by: sdpython <[email protected]>

* lint

Signed-off-by: sdpython <[email protected]>

* add the possibility to test custom operators

Signed-off-by: sdpython <[email protected]>

* add Compress

Signed-off-by: sdpython <[email protected]>

* add Concat

Signed-off-by: sdpython <[email protected]>

* add ConcatFromSequence

Signed-off-by: sdpython <[email protected]>

* add CumSum

Signed-off-by: sdpython <[email protected]>

* add ConstantOfShape, DepthToSpace, SpaceToDepth, Det, DequantizeLinear

Signed-off-by: sdpython <[email protected]>

* fix string issue

Signed-off-by: sdpython <[email protected]>

* add DFT

Signed-off-by: sdpython <[email protected]>

* add Div, DropOut

Signed-off-by: sdpython <[email protected]>

* add Einsum, Elu, Equal, Erf

Signed-off-by: sdpython <[email protected]>

* add Expand, EyeLike, Flatten, Gather, GatherElements, GatherND

Signed-off-by: sdpython <[email protected]>

* add Floor, Gemm, GlobalAveragePool, GlobalMaxPool

Signed-off-by: sdpython <[email protected]>

* add GRU, HardSigmoid, Hardmax

Signed-off-by: sdpython <[email protected]>

* lint

Signed-off-by: sdpython <[email protected]>

* add IsInf, Isnan, LayerNormalization, LeakyRelu, Less, LessOrEqual

Signed-off-by: sdpython <[email protected]>

* add Loop, LpNormalization, LogSoftmax, Softmax, LRN, LSTM

Signed-off-by: sdpython <[email protected]>

* add unit test on Loop

Signed-off-by: sdpython <[email protected]>

* add Slice, Unsqueeze

Signed-off-by: sdpython <[email protected]>

* lint

Signed-off-by: sdpython <[email protected]>

* complete Loop

Signed-off-by: sdpython <[email protected]>

* sort test

Signed-off-by: sdpython <[email protected]>

* fix a couple of issues

Signed-off-by: sdpython <[email protected]>

* add Constant_1

Signed-off-by: sdpython <[email protected]>

* add Where

Signed-off-by: sdpython <[email protected]>

* add Max, Mean, Min, Mod

Signed-off-by: sdpython <[email protected]>

* add Not, OneHot, NonZero, NLL

Signed-off-by: sdpython <[email protected]>

* add Optional

Signed-off-by: sdpython <[email protected]>

* remove unnecessary constructor

Signed-off-by: sdpython <[email protected]>

* add Pad, Pow, PRelu

Signed-off-by: sdpython <[email protected]>

* fix pad

Signed-off-by: sdpython <[email protected]>

* lint

Signed-off-by: sdpython <[email protected]>

* add reduce operators

Signed-off-by: sdpython <[email protected]>

* add quantize, range, reciprocal

Signed-off-by: sdpython <[email protected]>

* add dynamic quantize

Signed-off-by: sdpython <[email protected]>

* add RandomUniform

Signed-off-by: sdpython <[email protected]>

* add random

Signed-off-by: sdpython <[email protected]>

* add Trilu, Selu

Signed-off-by: sdpython <[email protected]>

* lint

Signed-off-by: sdpython <[email protected]>

* bugs

Signed-off-by: sdpython <[email protected]>

* add soft...3

Signed-off-by: sdpython <[email protected]>

* add squeeze

Signed-off-by: sdpython <[email protected]>

* add Resize

Signed-off-by: sdpython <[email protected]>

* add window functions

Signed-off-by: sdpython <[email protected]>

* lint

Signed-off-by: sdpython <[email protected]>

* add Scatter

Signed-off-by: sdpython <[email protected]>

* add RNN

Signed-off-by: sdpython <[email protected]>

* lint

Signed-off-by: sdpython <[email protected]>

* add last python ops

Signed-off-by: sdpython <[email protected]>

* lint

Signed-off-by: sdpython <[email protected]>

* lint

Signed-off-by: sdpython <[email protected]>

* add domaine ai.onnx.preview.training

Signed-off-by: sdpython <[email protected]>

* lint

Signed-off-by: sdpython <[email protected]>

* lint

Signed-off-by: sdpython <[email protected]>

* type

Signed-off-by: xadupre <[email protected]>

* lint

Signed-off-by: xadupre <[email protected]>

* add ReverseSequence

Signed-off-by: xadupre <[email protected]>

* lint

Signed-off-by: xadupre <[email protected]>

* lint

Signed-off-by: xadupre <[email protected]>

* fix reverse sequence

Signed-off-by: xadupre <[email protected]>

* add tile

Signed-off-by: xadupre <[email protected]>

* add SequenceMap

Signed-off-by: xadupre <[email protected]>

* fix sequence operators

Signed-off-by: xadupre <[email protected]>

* add operator scan

Signed-off-by: xadupre <[email protected]>

* add op MelWeightMatrix

Signed-off-by: xadupre <[email protected]>

* lint

Signed-off-by: xadupre <[email protected]>

* add MatMulInteger

Signed-off-by: xadupre <[email protected]>

* add Optional

Signed-off-by: xadupre <[email protected]>

* add instance normalization

Signed-off-by: xadupre <[email protected]>

* first step with standard functions

Signed-off-by: xadupre <[email protected]>

* implement method eval

Signed-off-by: xadupre <[email protected]>

* lint

Signed-off-by: xadupre <[email protected]>

* lint

Signed-off-by: xadupre <[email protected]>

* overriden

Signed-off-by: xadupre <[email protected]>

* modified

Signed-off-by: xadupre <[email protected]>

* attributes

Signed-off-by: xadupre <[email protected]>

* extend support for linked attributes

Signed-off-by: xadupre <[email protected]>

* support linked attributes

Signed-off-by: xadupre <[email protected]>

* remove schema_domain

Signed-off-by: xadupre <[email protected]>

* add conv

Signed-off-by: xadupre <[email protected]>

* finalize conv

Signed-off-by: xadupre <[email protected]>

* add ConvInteger

Signed-off-by: xadupre <[email protected]>

* add CropAndPad

Signed-off-by: xadupre <[email protected]>

* lint

Signed-off-by: xadupre <[email protected]>

* add qlinearconv

Signed-off-by: xadupre <[email protected]>

* add im2col

Signed-off-by: xadupre <[email protected]>

* update im2col

Signed-off-by: xadupre <[email protected]>

* fix im2col

Signed-off-by: xadupre <[email protected]>

* lint

Signed-off-by: xadupre <[email protected]>

* lint

Signed-off-by: xadupre <[email protected]>

* add col2im

Signed-off-by: xadupre <[email protected]>

* replace numpy. by np.

Signed-off-by: xadupre <[email protected]>

* complete replace numpy into np

Signed-off-by: xadupre <[email protected]>

* List[List|np.ndarray]

Signed-off-by: xadupre <[email protected]>

* review comments

Signed-off-by: xadupre <[email protected]>

* refactor unit test

Signed-off-by: xadupre <[email protected]>

* fix col2im

Signed-off-by: xadupre <[email protected]>

* lint

Signed-off-by: xadupre <[email protected]>

* lint

Signed-off-by: xadupre <[email protected]>

* add conv_transpose

Signed-off-by: xadupre <[email protected]>

* add convtranspose

Signed-off-by: xadupre <[email protected]>

* finalize ConvTranspose

Signed-off-by: xadupre <[email protected]>

* modif

Signed-off-by: xadupre <[email protected]>

* add grid_sample

Signed-off-by: xadupre <[email protected]>

* add max pool

Signed-off-by: xadupre <[email protected]>

* one move for max_pool

Signed-off-by: xadupre <[email protected]>

* raise exception when not implemented

Signed-off-by: xadupre <[email protected]>

* add maxunpool

Signed-off-by: xadupre <[email protected]>

* lint

Signed-off-by: xadupre <[email protected]>

* lint

Signed-off-by: xadupre <[email protected]>

* lint

Signed-off-by: xadupre <[email protected]>

* add non max suppression

Signed-off-by: xadupre <[email protected]>

* add has_attribute

Signed-off-by: xadupre <[email protected]>

* add max_pool_2d

Signed-off-by: xadupre <[email protected]>

* lint

Signed-off-by: xadupre <[email protected]>

* merge from main branch

Signed-off-by: xadupre <[email protected]>

* add roi align

Signed-off-by: xadupre <[email protected]>

* add RoiAlign

Signed-off-by: xadupre <[email protected]>

* lint

Signed-off-by: xadupre <[email protected]>

* simplifies overriden attributes

Signed-off-by: xadupre <[email protected]>

* lint

Signed-off-by: xadupre <[email protected]>

* lint

Signed-off-by: xadupre <[email protected]>

* tfidf

Signed-off-by: xadupre <[email protected]>

* tfidf

Signed-off-by: xadupre <[email protected]>

* fix tfidfvectorizer

Signed-off-by: xadupre <[email protected]>

* improve tfidf

Signed-off-by: xadupre <[email protected]>

* lint

Signed-off-by: xadupre <[email protected]>

* rename runtime into fn

Signed-off-by: xadupre <[email protected]>

* rename into funconnx

Signed-off-by: xadupre <[email protected]>

* rename Inference into ProtoRun

Signed-off-by: xadupre <[email protected]>

* fix Tfidf

Signed-off-by: xadupre <[email protected]>

* style

Signed-off-by: xadupre <[email protected]>

* add bitwise_not

Signed-off-by: xadupre <[email protected]>

* documentation

Signed-off-by: xadupre <[email protected]>

* add operator upsample

Signed-off-by: xadupre <[email protected]>

* improve code

Signed-off-by: xadupre <[email protected]>

* lint

Signed-off-by: xadupre <[email protected]>

* remove one warning

Signed-off-by: xadupre <[email protected]>

* support bfloat16 in cast operation

Signed-off-by: xadupre <[email protected]>

* refactoring of backend test

Signed-off-by: xadupre <[email protected]>

* add annotation

Signed-off-by: xadupre <[email protected]>

* lint

Signed-off-by: xadupre <[email protected]>

* lint

Signed-off-by: xadupre <[email protected]>

* improve support

Signed-off-by: xadupre <[email protected]>

* add more tests

Signed-off-by: xadupre <[email protected]>

* add SequenceErase

Signed-off-by: xadupre <[email protected]>

* lint

Signed-off-by: xadupre <[email protected]>

* lint

Signed-off-by: xadupre <[email protected]>

* lint

Signed-off-by: xadupre <[email protected]>

* lint

Signed-off-by: xadupre <[email protected]>

* fix bugs in Conv

Signed-off-by: xadupre <[email protected]>

* fix verbose test

Signed-off-by: xadupre <[email protected]>

* implement group for conv

Signed-off-by: xadupre <[email protected]>

* add conv 3d

Signed-off-by: xadupre <[email protected]>

* more comments

Signed-off-by: xadupre <[email protected]>

* complete max pool

Signed-off-by: xadupre <[email protected]>

* lint

Signed-off-by: xadupre <[email protected]>

* resize

Signed-off-by: xadupre <[email protected]>

* roi

Signed-off-by: xadupre <[email protected]>

* fix rnn, gru

Signed-off-by: xadupre <[email protected]>

* improve stft, still not accurate

Signed-off-by: xadupre <[email protected]>

* rename into reference

Signed-off-by: xadupre <[email protected]>

* remove unnecessary type ignore

Signed-off-by: xadupre <[email protected]>

* reorder import

Signed-off-by: xadupre <[email protected]>

* loop

Signed-off-by: xadupre <[email protected]>

* lint

Signed-off-by: xadupre <[email protected]>

* use absolute import

Signed-off-by: xadupre <[email protected]>

* import

Signed-off-by: xadupre <[email protected]>

* rename into ReferenceEvaluator

Signed-off-by: xadupre <[email protected]>

* rename files

Signed-off-by: xadupre <[email protected]>

* fix conv

Signed-off-by: xadupre <[email protected]>

* lint

Signed-off-by: xadupre <[email protected]>

* fix maxpool

Signed-off-by: xadupre <[email protected]>

* improves scatter elements

Signed-off-by: xadupre <[email protected]>

* remove duplicated function bfloat16_to_float32

Signed-off-by: xadupre <[email protected]>

* annotation

Signed-off-by: xadupre <[email protected]>

* remove pprint

Signed-off-by: xadupre <[email protected]>

* lint

Signed-off-by: xadupre <[email protected]>

* fix ConvTranspose

Signed-off-by: xadupre <[email protected]>

* improves conv transpose

Signed-off-by: xadupre <[email protected]>

* black

Signed-off-by: xadupre <[email protected]>

* fix pad in ConvTranspose

Signed-off-by: xadupre <[email protected]>

* fix convtranspose

Signed-off-by: xadupre <[email protected]>

* better documentation for bfloat16_to_float32

Signed-off-by: xadupre <[email protected]>

* extend annotation

Signed-off-by: xadupre <[email protected]>

* improves resize

Signed-off-by: xadupre <[email protected]>

* fix Resize

Signed-off-by: xadupre <[email protected]>

* complete momentum

Signed-off-by: xadupre <[email protected]>

* fix stft

Signed-off-by: xadupre <[email protected]>

* better comment

Signed-off-by: xadupre <[email protected]>

* update the number of mandatory tests to pass

Signed-off-by: xadupre <[email protected]>

* documentation

Signed-off-by: xadupre <[email protected]>

* merge

Signed-off-by: xadupre <[email protected]>

* fix context dependent implementation

Signed-off-by: xadupre <[email protected]>

* lint

Signed-off-by: xadupre <[email protected]>

* lint

Signed-off-by: xadupre <[email protected]>

* misspelling

Signed-off-by: xadupre <[email protected]>

Signed-off-by: xadupre <[email protected]>
Signed-off-by: sdpython <[email protected]>
Co-authored-by: sdpython <[email protected]>
Signed-off-by: Yi Zhang <[email protected]>

* fix parse repo name

Signed-off-by: Yi Zhang <[email protected]>

* revert change

Signed-off-by: Yi Zhang <[email protected]>

Signed-off-by: Chun-Wei Chen <[email protected]>
Signed-off-by: Yi Zhang <[email protected]>
Signed-off-by: Christian Bourjau <[email protected]>
Signed-off-by: xadupre <[email protected]>
Signed-off-by: sdpython <[email protected]>
Co-authored-by: Chun-Wei Chen <[email protected]>
Co-authored-by: Christian Bourjau <[email protected]>
Co-authored-by: G. Ramalingam <[email protected]>
Co-authored-by: Xavier Dupré <[email protected]>
Co-authored-by: sdpython <[email protected]>

onnx/hub.py

@@ -89,10 +89,10 @@ def _parse_repo_info(repo: str) -> Tuple[str, str, str]:
     """
     Gets the repo owner, name and ref from a repo specification string.
     """
-    repo_owner = repo.split("/")[0]
-    repo_name = repo.split("/")[1].split(":")[0]
+    repo_owner = repo.split(":")[0].split("/")[0]
+    repo_name = repo.split(":")[0].split("/")[1]
     if ":" in repo:
-        repo_ref = repo.split("/")[1].split(":")[1]
+        repo_ref = repo.split(":")[1]
     else:
         repo_ref = "main"
     return repo_owner, repo_name, repo_ref