From 4799d5ce578553eec54e737c74cb60cfa07066db Mon Sep 17 00:00:00 2001
From: Fischer Moseley <42497969+fischermoseley@users.noreply.github.com>
Date: Mon, 7 Oct 2024 20:58:18 -0700
Subject: [PATCH] logic_analyzer: tidy some formatting/typos from rebase

---
 src/manta/logic_analyzer/__init__.py |  2 --
 src/manta/logic_analyzer/capture.py  | 11 ++++++-----
 2 files changed, 6 insertions(+), 7 deletions(-)

diff --git a/src/manta/logic_analyzer/__init__.py b/src/manta/logic_analyzer/__init__.py
index fe69420..b566421 100644
--- a/src/manta/logic_analyzer/__init__.py
+++ b/src/manta/logic_analyzer/__init__.py
@@ -6,8 +6,6 @@ from manta.logic_analyzer.trigger_block import LogicAnalyzerTriggerBlock
 from manta.memory_core import MemoryCore
 from manta.utils import *
 
-import math
-
 
 class LogicAnalyzerCore(MantaCore):
     """
diff --git a/src/manta/logic_analyzer/capture.py b/src/manta/logic_analyzer/capture.py
index bde59d1..5a5e2eb 100644
--- a/src/manta/logic_analyzer/capture.py
+++ b/src/manta/logic_analyzer/capture.py
@@ -1,3 +1,8 @@
+import math
+from datetime import datetime
+
+from vcd import VCDWriter
+
 from manta.logic_analyzer.fsm import TriggerModes
 from manta.logic_analyzer.playback import LogicAnalyzerPlayback
 
@@ -96,10 +101,6 @@ class LogicAnalyzerCapture:
             None
         """
 
-        from datetime import datetime
-
-        from vcd import VCDWriter
-
         # Compute the timescale from the frequency of the provided clock
         half_period = 1 / (2 * self._interface._clock_freq)
         exponent = math.floor(math.log10(half_period))
@@ -158,7 +159,7 @@ class LogicAnalyzerCapture:
                 # Set the trigger (if there is one)
                 if self._trigger_mode == TriggerModes.SINGLE_SHOT:
                     triggered = (sample_index // 2) >= self._trigger_location
-                    writer.change(trigger, sample_timestep, triggered)
+                    writer.change(trigger, sample_timestamp, triggered)
 
                 # Add other signals
                 for signal in signals: