From 6d171d470395638e2934bfbb6478484fc360ea26 Mon Sep 17 00:00:00 2001
From: Carl Philipp Klemm <philipp@uvos.xyz>
Date: Mon, 9 Oct 2023 15:53:07 +0200
Subject: [PATCH] resistor: fix script not returning torch.complex in scrip
calculation
---
model.cpp | 2 +-
resistor.cpp | 2 +-
2 files changed, 2 insertions(+), 2 deletions(-)
diff --git a/model.cpp b/model.cpp
index de58e09..ce60208 100644
--- a/model.cpp
+++ b/model.cpp
@@ -620,7 +620,7 @@ std::string Model::getTorchScript()
out<<" assert parameters.size(0) is "<<parameters.size()<<"\n\n";
for(size_t i = 0; i < parameters.size(); ++i)
out<<" "<<parameters[i]<<" = parameters["<<i<<"]\n";
- out<<"\n return "<<formular<<'\n';
+ out<<"\n return "<<formular<<"+0*omegas\n";
return out.str();
}
diff --git a/resistor.cpp b/resistor.cpp
index 2318555..9681714 100644
--- a/resistor.cpp
+++ b/resistor.cpp
@@ -58,5 +58,5 @@ std::string Resistor::getTorchScript(std::vector<std::string>& parameters)
{
parameters.push_back(getUniqueName() + "_0");
- return parameters.back();
+ return parameters.back() + "*(1+0j)";
}
--
GitLab