type annotations

finch-tensor · Dec 6, 2023 · 0b10a71 · 0b10a71
1 parent 17b6429
commit 0b10a71
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/ext/HDF5Ext.jl b/ext/HDF5Ext.jl
@@ -6,13 +6,13 @@ using Finch.DataStructures
 
 isdefined(Base, :get_extension) ? (using HDF5) : (using ..HDF5)
 
-function Finch.bspread_h5(fname)
+function Finch.bspread_h5(fname::AbstractString)
     h5open(fname, "r") do io
         Finch.bspread(io)
     end
 end
 
-function Finch.bspwrite_h5(fname, arr, attrs = OrderedDict())
+function Finch.bspwrite_h5(fname::AbstractString, arr, attrs = OrderedDict())
     h5open(fname, "w") do io
         Finch.bspwrite(io, arr, attrs)
     end

diff --git a/ext/NPZExt.jl b/ext/NPZExt.jl
@@ -28,11 +28,11 @@ Finch.bspwrite_header(g::NPYPath, str::String) = write(joinpath(mkpath(g.dirname
 Finch.bspread_vector(g::NPYPath, key) = g[key]
 Finch.bspwrite_vector(g::NPYPath, vec, key) = (g[key] = vec)
 
-function Finch.bspread_bspnpy(fname)
+function Finch.bspread_bspnpy(fname::AbstractString)
     bspread(NPYPath(fname))
 end
 
-function Finch.bspwrite_bspnpy(fname, arr, attrs = OrderedDict())
+function Finch.bspwrite_bspnpy(fname::AbstractString, arr, attrs = OrderedDict())
     bspwrite(NPYPath(fname), arr, attrs)
     fname
 end