fix: doctests

amazon-braket · Sep 27, 2024 · b8867f4 · b8867f4
1 parent d4a0f36
commit b8867f4
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 5 deletions.
diff --git a/.github/workflows/CI.yml b/.github/workflows/CI.yml
@@ -170,5 +170,5 @@ jobs:
         run: |
           using Documenter: DocMeta, doctest
           using Braket
-          DocMeta.setdocmeta!(Braket, :DocTestSetup, :(using Braket); recursive=true)
+          DocMeta.setdocmeta!(Braket, :DocTestSetup, :(using Braket, Braket.Dates); recursive=true)
           doctest(Braket)
diff --git a/docs/make.jl b/docs/make.jl
@@ -1,6 +1,6 @@
 push!(LOAD_PATH,"../src/")
 
-using Documenter, Braket 
+using Documenter, Braket, Braket.Dates
 
 makedocs(sitename="Braket.jl")
 

diff --git a/src/circuit.jl b/src/circuit.jl
@@ -374,7 +374,7 @@ julia> circ.instructions
  Braket.Instruction{Reset}(Reset(), QubitSet(0))
 ```
 """
-Base.reset(c::Circuit, target_qubits) = foreach(t->add_instruction!(c, Instruction(Reset(), t)), target_qubits)
+Base.reset(c::Circuit, target_qubits) = (foreach(t->add_instruction!(c, Instruction(Reset(), t)), target_qubits); return c)
 
 """
     delay(c::Circuit, duration::Dates.Period, target_qubits) -> Circuit
@@ -395,7 +395,7 @@ julia> circ.instructions
  Braket.Instruction{Delay}(Delay(Nanosecond(10)), QubitSet(0, 1))
 ```
 """
-delay(c::Circuit, duration::Dates.Period, target_qubits) = add_instruction!(c, Instruction(Delay(duration), target_qubits))
+delay(c::Circuit, duration::Dates.Period, target_qubits) = (add_instruction!(c, Instruction(Delay(duration), target_qubits)); return c)
 # TODO enforce this in `Moments` as well
 
 """
@@ -417,7 +417,7 @@ julia> circ.instructions
  Braket.Instruction{Barrier}(Barrier(), QubitSet(0, 1))
 ```
 """
-barrier(c::Circuit, target_qubits) = add_instruction!(c, Instruction(Barrier(), target_qubits))
+barrier(c::Circuit, target_qubits) = (add_instruction!(c, Instruction(Barrier(), target_qubits)); return c)
 
 function openqasm_header(c::Circuit, sps::SerializationProperties=OpenQASMSerializationProperties())
     ir_instructions = ["OPENQASM 3.0;"]