Skip to content
Snippets Groups Projects
Commit 6d171d47 authored by Carl Philipp Klemm's avatar Carl Philipp Klemm
Browse files

resistor: fix script not returning torch.complex in scrip calculation

parent 4a0dc913
Branches
No related tags found
No related merge requests found
......@@ -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();
}
......
......@@ -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)";
}
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment