Fix tests

ITensor · May 8, 2024 · a34a359 · a34a359
1 parent 6a7db53
commit a34a359
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/ITensorGPU.jl b/src/ITensorGPU.jl
@@ -16,7 +16,7 @@ export cuITensor, randomCuITensor
 
 # TODO: Change over to `using ITensorMPS`
 # once it is registered.
-using ITensors.ITensorMPS: MPO, MPS, randomMPS
+using ITensors.ITensorMPS: MPO, MPS, randomMPO, randomMPS
 function cuMPS(args...; kwargs...)
   return adapt(CuArray, MPS(args...; kwargs...))
 end

diff --git a/test/test_cudiag.jl b/test/test_cudiag.jl
@@ -30,7 +30,7 @@ using Test: @test, @test_broken, @testset
       @test_broken collect(CuArray(C)) ≈ collect(CuMatrix(Aij, j, i)) * diagm(0 => Dv)
     end
     @testset "Test contract cuDiagITensors (Diag*Diag -> Diag)" begin
-      @test_brokrn E * D
+      @test_broken E * D
       ## C = E * D
       ## cC = CuArray(C)
       ## @test collect(cC) ≈ diagm(0 => Ev) * diagm(0 => Dv)