Mercurial > repos > public > sbplib_julia

changeset 1231:de6a9635f293 refactor/LazyTensors/tuple_manipulation

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Add docstring for concatenate_tuples
author Jonatan Werpers <jonatan@werpers.com>
date Sun, 19 Feb 2023 12:14:43 +0100
parents 723a583cef96
children a8fa8c1137cc 66012b95fa7b
files src/LazyTensors/tuple_manipulation.jl test/LazyTensors/tuple_manipulation_test.jl
diffstat 2 files changed, 6 insertions(+), 2 deletions(-) [+]
line wrap: on
line diff
--- a/src/LazyTensors/tuple_manipulation.jl	Sun Feb 19 11:45:48 2023 +0100
+++ b/src/LazyTensors/tuple_manipulation.jl	Sun Feb 19 12:14:43 2023 +0100
@@ -56,6 +56,12 @@
     return ntuple(i->cum_szs[i]+1:cum_szs[i+1], length(szs))
 end
 
+
+"""
+    concatenate_tuples(t...)
+
+Concatenate tuples.
+"""
 concatenate_tuples(t::Tuple,ts::Vararg{Tuple}) = (t..., concatenate_tuples(ts...)...)
 concatenate_tuples(t::Tuple) = t
 
--- a/test/LazyTensors/tuple_manipulation_test.jl	Sun Feb 19 11:45:48 2023 +0100
+++ b/test/LazyTensors/tuple_manipulation_test.jl	Sun Feb 19 12:14:43 2023 +0100
@@ -34,9 +34,7 @@
         @test_throws ArgumentError(err_msg) LazyTensors.split_tuple((1,2), (1,2))
         @test_throws ArgumentError(err_msg) LazyTensors.split_tuple((1,2), (1))
 
-
         split_tuple_static(t, ::Val{SZS}) where {SZS} = LazyTensors.split_tuple(t,SZS)
-
         @inferred split_tuple_static((1,2,3,4,5,6), Val((3,1,2)))
         @inferred split_tuple_static((1,2,3,4),Val((3,1)))
         @inferred split_tuple_static((1,2,true,4),Val((3,1)))