pyedb 0.13.0__py3-none-any.whl → 0.14.0__py3-none-any.whl

pyedb/dotnet/edb_core/components.py

@@ -30,8 +30,8 @@ import re
 import warnings
 
 from pyedb.dotnet.clr_module import String
+from pyedb.dotnet.edb_core.cell.hierarchy.component import EDBComponent
 from pyedb.dotnet.edb_core.definition.component_def import EDBComponentDef
-from pyedb.dotnet.edb_core.edb_data.components_data import EDBComponent
 from pyedb.dotnet.edb_core.edb_data.padstacks_data import EDBPadstackInstance
 from pyedb.dotnet.edb_core.edb_data.sources import Source, SourceType
 from pyedb.dotnet.edb_core.general import convert_py_list_to_net_list
@@ -40,7 +40,6 @@ from pyedb.generic.general_methods import (
     _retry_ntimes,
     generate_unique_name,
     get_filename_without_extension,
-    pyedb_function_handler,
 )
 from pyedb.modeler.geometry_operators import GeometryOperators
 
@@ -85,7 +84,6 @@ class Components(object):
     >>> edbapp.components
     """
 
-    @pyedb_function_handler()
     def __getitem__(self, name):
         """Get  a component or component definition from the Edb project.
 
@@ -95,7 +93,7 @@ class Components(object):
 
         Returns
         -------
-        :class:`pyedb.dotnet.edb_core.edb_data.components_data.EDBComponent`
+        :class:`pyedb.dotnet.edb_core.cell.hierarchy.component.EDBComponent`
 
         """
         if name in self.instances:
@@ -128,7 +126,6 @@ class Components(object):
     def _edb(self):
         return self._pedb.edb_api
 
-    @pyedb_function_handler()
     def _init_parts(self):
         a = self.components
         a = self.resistors
@@ -171,7 +168,7 @@ class Components(object):
 
         Returns
         -------
-        dict[str, :class:`pyedb.dotnet.edb_core.edb_data.components_data.EDBComponent`]
+        dict[str, :class:`pyedb.dotnet.edb_core.cell.hierarchy.component.EDBComponent`]
             Default dictionary for the EDB component.
 
         Examples
@@ -191,7 +188,7 @@ class Components(object):
 
         Returns
         -------
-        Dict[str, :class:`pyedb.dotnet.edb_core.edb_data.components_data.EDBComponent`]
+        Dict[str, :class:`pyedb.dotnet.edb_core.cell.hierarchy.component.EDBComponent`]
             Default dictionary for the EDB component.
 
         Examples
@@ -221,7 +218,6 @@ class Components(object):
         m = "Ansys.Ansoft.Edb.Definition.NPortComponentModel"
         return {name: l for name, l in self.definitions.items() if m in [i.ToString() for i in l._comp_model]}
 
-    @pyedb_function_handler()
     def import_definition(self, file_path):
         """Import component definition from json file.
 
@@ -256,7 +252,6 @@ class Components(object):
                         pass
         return True
 
-    @pyedb_function_handler()
     def export_definition(self, file_path):
         """Export component definitions to json file.
 
@@ -307,7 +302,6 @@ class Components(object):
             json.dump(data, f, ensure_ascii=False, indent=4)
         return file_path
 
-    @pyedb_function_handler()
     def refresh_components(self):
         """Refresh the component dictionary."""
         # self._logger.info("Refreshing the Components dictionary.")
@@ -324,7 +318,7 @@ class Components(object):
 
         Returns
         -------
-        dict[str, :class:`pyedb.dotnet.edb_core.edb_data.components_data.EDBComponent`]
+        dict[str, :class:`pyedb.dotnet.edb_core.cell.hierarchy.component.EDBComponent`]
             Dictionary of resistors.
 
         Examples
@@ -346,7 +340,7 @@ class Components(object):
 
         Returns
         -------
-        dict[str, :class:`pyedb.dotnet.edb_core.edb_data.components_data.EDBComponent`]
+        dict[str, :class:`pyedb.dotnet.edb_core.cell.hierarchy.component.EDBComponent`]
             Dictionary of capacitors.
 
         Examples
@@ -368,7 +362,7 @@ class Components(object):
 
         Returns
         -------
-        dict[str, :class:`pyedb.dotnet.edb_core.edb_data.components_data.EDBComponent`]
+        dict[str, :class:`pyedb.dotnet.edb_core.cell.hierarchy.component.EDBComponent`]
             Dictionary of inductors.
 
         Examples
@@ -391,7 +385,7 @@ class Components(object):
 
         Returns
         -------
-        dict[str, :class:`pyedb.dotnet.edb_core.edb_data.components_data.EDBComponent`]
+        dict[str, :class:`pyedb.dotnet.edb_core.cell.hierarchy.component.EDBComponent`]
             Dictionary of integrated circuits.
 
         Examples
@@ -414,7 +408,7 @@ class Components(object):
 
         Returns
         -------
-        dict[str, :class:`pyedb.dotnet.edb_core.edb_data.components_data.EDBComponent`]
+        dict[str, :class:`pyedb.dotnet.edb_core.cell.hierarchy.component.EDBComponent`]
             Dictionary of circuit inputs and outputs.
 
         Examples
@@ -437,7 +431,7 @@ class Components(object):
 
         Returns
         -------
-        dict[str, :class:`pyedb.dotnet.edb_core.edb_data.components_data.EDBComponent`]
+        dict[str, :class:`pyedb.dotnet.edb_core.cell.hierarchy.component.EDBComponent`]
             Dictionary of other core components.
 
         Examples
@@ -479,7 +473,6 @@ class Components(object):
                 self._comps_by_part[val.partname] = [val]
         return self._comps_by_part
 
-    @pyedb_function_handler()
     def get_component_by_name(self, name):
         """Retrieve a component by name.
 
@@ -500,7 +493,6 @@ class Components(object):
         else:
             pass
 
-    @pyedb_function_handler()
     def get_components_from_nets(self, netlist=None):
         """Retrieve components from a net list.
 
@@ -525,7 +517,6 @@ class Components(object):
                 cmp_list.append(refdes)
         return cmp_list
 
-    @pyedb_function_handler()
     def _get_edb_pin_from_pin_name(self, cmp, pin):
         if not isinstance(cmp, self._pedb.edb_api.cell.hierarchy.component):
             return False
@@ -536,7 +527,6 @@ class Components(object):
             return pins[0]
         return False
 
-    @pyedb_function_handler()
     def get_component_placement_vector(
         self,
         mounted_component,
@@ -634,7 +624,6 @@ class Components(object):
         self._logger.warning("Failed to compute vector.")
         return False, [0, 0], 0, 0
 
-    @pyedb_function_handler()
     def get_solder_ball_height(self, cmp):
         """Get component solder ball height.
 
@@ -656,7 +645,6 @@ class Components(object):
             return cmp_prop.GetSolderBallProperty().GetHeight()
         return False
 
-    @pyedb_function_handler()
     def create_source_on_component(self, sources=None):
         """Create voltage, current source, or resistor on component.
 
@@ -735,7 +723,6 @@ class Components(object):
                 )
         return True
 
-    @pyedb_function_handler()
     def create_port_on_pins(self, refdes, pins, reference_pins, impedance=50.0, port_name=None, pec_boundary=False):
         """Create circuit port between pins and reference ones.
 
@@ -855,7 +842,6 @@ class Components(object):
             return term
         return False
 
-    @pyedb_function_handler()
     def create_port_on_component(
         self,
         component,
@@ -1030,7 +1016,6 @@ class Components(object):
                         self.create_port_on_pins(component, pin, ref_pins)
         return True
 
-    @pyedb_function_handler()
     def _create_terminal(self, pin, term_name=None):
         """Create terminal on component pin.
 
@@ -1060,7 +1045,6 @@ class Components(object):
         )
         return term
 
-    @pyedb_function_handler()
     def _get_closest_pin_from(self, pin, ref_pinlist):
         """Returns the closest pin from given pin among the list of reference pins.
 
@@ -1092,7 +1076,6 @@ class Components(object):
                 closest_pin = ref_pin
         return closest_pin
 
-    @pyedb_function_handler()
     def replace_rlc_by_gap_boundaries(self, component=None):
         """Replace RLC component by RLC gap boundaries. These boundary types are compatible with 3D modeler export.
         Only 2 pins RLC components are supported in this command.
@@ -1134,7 +1117,6 @@ class Components(object):
         component.is_enabled = False
         return self.add_rlc_boundary(component.refdes, False)
 
-    @pyedb_function_handler()
     def deactivate_rlc_component(self, component=None, create_circuit_port=False, pec_boundary=False):
         """Deactivate RLC component with a possibility to convert it to a circuit port.
 
@@ -1187,7 +1169,6 @@ class Components(object):
             component=component.refdes, circuit_ports=create_circuit_port, pec_boundary=pec_boundary
         )
 
-    @pyedb_function_handler()
     def add_port_on_rlc_component(self, component=None, circuit_ports=True, pec_boundary=False):
         """Deactivate RLC component and replace it with a circuit port.
         The circuit port supports only two-pin components.
@@ -1261,7 +1242,6 @@ class Components(object):
             return True
         return False
 
-    @pyedb_function_handler()
     def add_rlc_boundary(self, component=None, circuit_type=True):
         """Add RLC gap boundary on component and replace it with a circuit port.
         The circuit port supports only 2-pin components.
@@ -1335,7 +1315,6 @@ class Components(object):
             self._logger.info("Component {} has been replaced by port".format(component.refdes))
             return True
 
-    @pyedb_function_handler()
     def _create_pin_group_terminal(self, pingroup, isref=False, term_name=None, term_type="circuit"):
         """Creates an EDB pin group terminal from a given EDB pin group.
 
@@ -1370,7 +1349,6 @@ class Components(object):
             pingroup_term.SetIsAutoPort(True)
         return pingroup_term
 
-    @pyedb_function_handler()
     def _is_top_component(self, cmp):
         """Test the component placement layer.
 
@@ -1392,7 +1370,6 @@ class Components(object):
         else:
             return False
 
-    @pyedb_function_handler()
     def _getComponentDefinition(self, name, pins):
         componentDefinition = self._pedb.edb_api.definition.ComponentDef.FindByName(self._db, name)
         if componentDefinition.IsNull():
@@ -1415,7 +1392,6 @@ class Components(object):
             self._logger.warning("Found existing component definition for footprint {}".format(name))
         return componentDefinition
 
-    @pyedb_function_handler()
     def create_rlc_component(
         self, pins, component_name="", r_value=None, c_value=None, l_value=None, is_parallel=False
     ):  # pragma: no cover
@@ -1455,7 +1431,6 @@ class Components(object):
             is_parallel=is_parallel,
         )
 
-    @pyedb_function_handler()
     def create(
         self,
         pins,
@@ -1571,7 +1546,6 @@ class Components(object):
         self._cmp[new_cmp.GetName()] = new_edb_comp
         return new_edb_comp
 
-    @pyedb_function_handler()
     def create_component_from_pins(
         self, pins, component_name, placement_layer=None, component_part_name=None
     ):  # pragma: no cover
@@ -1614,7 +1588,6 @@ class Components(object):
             is_rlc=False,
         )
 
-    @pyedb_function_handler()
     def set_component_model(self, componentname, model_type="Spice", modelpath=None, modelname=None):
         """Assign a Spice or Touchstone model to a component.
 
@@ -1703,7 +1676,6 @@ class Components(object):
                 return False
         return True
 
-    @pyedb_function_handler()
     def create_pingroup_from_pins(self, pins, group_name=None):
         """Create a pin group on a component.
 
@@ -1771,7 +1743,6 @@ class Components(object):
             pingroup.SetNet(pins[0].GetNet())
             return pingroup
 
-    @pyedb_function_handler()
     def delete_single_pin_rlc(self, deactivate_only=False):
         # type: (bool) -> list
         """Delete all RLC components with a single pin.
@@ -1813,7 +1784,6 @@ class Components(object):
 
         return deleted_comps
 
-    @pyedb_function_handler()
     def delete_component(self, component_name):  # pragma: no cover
         """Delete a component.
 
@@ -1841,7 +1811,6 @@ class Components(object):
         warnings.warn("`delete_component` is deprecated. Use `delete` property instead.", DeprecationWarning)
         return self.delete(component_name=component_name)
 
-    @pyedb_function_handler()
     def delete(self, component_name):
         """Delete a component.
 
@@ -1871,7 +1840,6 @@ class Components(object):
             return True
         return False
 
-    @pyedb_function_handler()
     def disable_rlc_component(self, component_name):
         """Disable a RLC component.
 
@@ -1907,7 +1875,6 @@ class Components(object):
             return True
         return False
 
-    @pyedb_function_handler()
     def set_solder_ball(
         self,
         component="",
@@ -2014,7 +1981,6 @@ class Components(object):
         edb_cmp.SetComponentProperty(cmp_property)
         return True
 
-    @pyedb_function_handler()
     def set_component_rlc(
         self,
         componentname,
@@ -2099,7 +2065,6 @@ class Components(object):
         self._logger.info("RLC properties for Component %s has been assigned.", componentname)
         return True
 
-    @pyedb_function_handler()
     def update_rlc_from_bom(
         self,
         bom_file,
@@ -2169,7 +2134,6 @@ class Components(object):
                 self.components[comp].is_enabled = False
         return found
 
-    @pyedb_function_handler()
     def import_bom(
         self,
         bom_file,
@@ -2260,7 +2224,6 @@ class Components(object):
                 self.components[comp].is_enabled = False
         return True
 
-    @pyedb_function_handler()
     def export_bom(self, bom_file, delimiter=","):
         """Export Bom file from layout.
 
@@ -2291,7 +2254,6 @@ class Components(object):
                 f.writelines([delimiter.join([refdes, part_name, comp_type, value + "\n"])])
         return True
 
-    @pyedb_function_handler
     def find_by_reference_designator(self, reference_designator):
         """Find a component.
 
@@ -2303,7 +2265,6 @@ class Components(object):
         obj = self._pedb.edb_api.cell.hierarchy.component.FindByName(self._active_layout, reference_designator)
         return EDBComponent(self._pedb, obj)
 
-    @pyedb_function_handler()
     def get_pin_from_component(self, component, netName=None, pinName=None):
         """Retrieve the pins of a component.
 
@@ -2356,7 +2317,6 @@ class Components(object):
             pins = [p for p in list(component.LayoutObjs) if int(p.GetObjType()) == 1 and p.IsLayoutPin()]
         return pins
 
-    @pyedb_function_handler()
     def get_aedt_pin_name(self, pin):
         """Retrieve the pin name that is shown in AEDT.
 
@@ -2388,7 +2348,6 @@ class Components(object):
         name = str(name).strip("'")
         return name
 
-    @pyedb_function_handler
     def get_pins(self, reference_designator, net_name=None, pin_name=None):
         """Get component pins.
 
@@ -2416,7 +2375,6 @@ class Components(object):
 
         return pins
 
-    @pyedb_function_handler()
     def get_pin_position(self, pin):
         """Retrieve the pin position in meters.
 
@@ -2450,7 +2408,6 @@ class Components(object):
         )
         return [pin_xy.X.ToDouble(), pin_xy.Y.ToDouble()]
 
-    @pyedb_function_handler()
     def get_pins_name_from_net(self, net_name, pin_list=None):
         """Retrieve pins belonging to a net.
 
@@ -2485,7 +2442,6 @@ class Components(object):
                 pin_names.append(self.get_aedt_pin_name(pin))
         return pin_names
 
-    @pyedb_function_handler()
     def get_nets_from_pin_list(self, PinList):
         """Retrieve nets with one or more pins.
 
@@ -2512,7 +2468,6 @@ class Components(object):
             netlist.append(pin.GetNet().GetName())
         return list(set(netlist))
 
-    @pyedb_function_handler()
     def get_component_net_connection_info(self, refdes):
         """Retrieve net connection information.
 
@@ -2602,7 +2557,6 @@ class Components(object):
 
         return through_comp_list
 
-    @pyedb_function_handler()
     def short_component_pins(self, component_name, pins_to_short=None, width=1e-3):
         """Short pins of component with a trace.
 

pyedb/dotnet/edb_core/definition/component_def.py

@@ -24,7 +24,6 @@ import os
 
 from pyedb.dotnet.edb_core.definition.component_model import NPortComponentModel
 from pyedb.dotnet.edb_core.utilities.obj_base import ObjBase
-from pyedb.generic.general_methods import pyedb_function_handler
 
 
 class EDBComponentDef(ObjBase):
@@ -84,7 +83,7 @@ class EDBComponentDef(ObjBase):
         -------
         dict of :class:`EDBComponent`
         """
-        from pyedb.dotnet.edb_core.edb_data.components_data import EDBComponent
+        from pyedb.dotnet.edb_core.cell.hierarchy.component import EDBComponent
 
         comp_list = [
             EDBComponent(self._pedb, l)
@@ -94,7 +93,6 @@ class EDBComponentDef(ObjBase):
         ]
         return {comp.refdes: comp for comp in comp_list}
 
-    @pyedb_function_handler()
     def assign_rlc_model(self, res=None, ind=None, cap=None, is_parallel=False):
         """Assign RLC to all components under this part name.
 
@@ -114,7 +112,6 @@ class EDBComponentDef(ObjBase):
             comp.assign_rlc_model(res, ind, cap, is_parallel)
         return True
 
-    @pyedb_function_handler()
     def assign_s_param_model(self, file_path, model_name=None, reference_net=None):
         """Assign S-parameter to all components under this part name.
 
@@ -133,7 +130,6 @@ class EDBComponentDef(ObjBase):
             comp.assign_s_param_model(file_path, model_name, reference_net)
         return True
 
-    @pyedb_function_handler()
     def assign_spice_model(self, file_path, model_name=None):
         """Assign Spice model to all components under this part name.
 
@@ -171,11 +167,9 @@ class EDBComponentDef(ObjBase):
                 temp[edb_object.name] = edb_object
         return temp
 
-    @pyedb_function_handler
     def _add_component_model(self, value):
         self._edb_object.AddComponentModel(value._edb_object)
 
-    @pyedb_function_handler
     def add_n_port_model(self, fpath, name=None):
         if not name:
             name = os.path.splitext(os.path.basename(fpath)[0])
@@ -188,7 +182,6 @@ class EDBComponentDef(ObjBase):
 
         self._add_component_model(n_port_comp_model)
 
-    @pyedb_function_handler
     def create(self, name):
         cell_type = self._pedb.edb_api.cell.CellType.FootprintCell
         footprint_cell = self._pedb._active_cell.cell.Create(self._pedb.active_db, cell_type, name)

pyedb/dotnet/edb_core/definition/component_model.py

@@ -21,7 +21,6 @@
 # SOFTWARE.
 
 from pyedb.dotnet.edb_core.utilities.obj_base import ObjBase
-from pyedb.generic.general_methods import pyedb_function_handler
 
 
 class ComponentModel(ObjBase):
@@ -31,7 +30,6 @@ class ComponentModel(ObjBase):
         super().__init__(pedb, edb_object)
         self._model_type_mapping = {"PinPairModel": self._pedb.edb_api.cell}
 
-    @pyedb_function_handler()
     def name(self):
         """Name of the component model."""
         return self._edb_object.GetName()

pyedb/dotnet/edb_core/definition/definitions.py

@@ -22,7 +22,6 @@
 
 from pyedb.dotnet.edb_core.definition.component_def import EDBComponentDef
 from pyedb.dotnet.edb_core.definition.package_def import PackageDef
-from pyedb.generic.general_methods import pyedb_function_handler
 
 
 class Definitions:
@@ -39,7 +38,6 @@ class Definitions:
         """Package definitions."""
         return {l.GetName(): PackageDef(self._pedb, l) for l in list(self._pedb.active_db.PackageDefs)}
 
-    @pyedb_function_handler
     def add_package_def(self, name, component_part_name=None, boundary_points=None):
         """Add a package definition.
 

pyedb/dotnet/edb_core/definition/package_def.py

@@ -22,7 +22,7 @@
 
 from pyedb.dotnet.edb_core.geometry.polygon_data import PolygonData
 from pyedb.dotnet.edb_core.utilities.obj_base import ObjBase
-from pyedb.generic.general_methods import pyedb_function_handler
+from pyedb.edb_logger import pyedb_logger
 
 
 class PackageDef(ObjBase):
@@ -37,7 +37,7 @@ class PackageDef(ObjBase):
         component_part_name : str, optional
         Part name of the component.
     extent_bounding_box : list, optional
-        Bounding box defines the shape of the package. For example, [[0, 0], ["2mm", "2mm"].
+        Bounding box defines the shape of the package. For example, [[0, 0], ["2mm", "2mm"]].
 
     """
 
@@ -48,7 +48,6 @@ class PackageDef(ObjBase):
         else:
             self._edb_object = edb_object
 
-    @pyedb_function_handler
     def __create_from_name(self, name, component_part_name=None, extent_bounding_box=None):
         """Create a package definition.
 
@@ -72,12 +71,16 @@ class PackageDef(ObjBase):
             bbox = [[y_pt1 - y_mid, x_pt1 - x_mid], [y_pt2 - y_mid, x_pt2 - x_mid]]
         else:
             bbox = extent_bounding_box
+        if bbox is None:
+            pyedb_logger.warning(
+                "Package creation uses bounding box but it cannot be inferred. "
+                "Please set argument 'component_part_name' or 'extent_bounding_box'."
+            )
         polygon_data = PolygonData(self._pedb, create_from_bounding_box=True, points=bbox)
 
         edb_object.SetExteriorBoundary(polygon_data._edb_object)
         return edb_object
 
-    @pyedb_function_handler
     def delete(self):
         """Delete a package definition object from the database."""
         return self._edb_object.Delete()
@@ -141,7 +144,6 @@ class PackageDef(ObjBase):
         value = self._pedb.edb_value(value)
         self._edb_object.SetHeight(value)
 
-    @pyedb_function_handler
     def set_heatsink(self, fin_base_height, fin_height, fin_orientation, fin_spacing, fin_thickness):
         from pyedb.dotnet.edb_core.utilities.heatsink import HeatSink
 

pyedb/dotnet/edb_core/edb_data/control_file.py

@@ -32,7 +32,6 @@ from pyedb.generic.general_methods import (
     env_value,
     is_ironpython,
     is_linux,
-    pyedb_function_handler,
 )
 from pyedb.misc.aedtlib_personalib_install import write_pretty_xml
 from pyedb.misc.misc import list_installed_ansysem
@@ -43,7 +42,6 @@ else:
     import subprocess
 
 
-@pyedb_function_handler()
 def convert_technology_file(tech_file, edbversion=None, control_file=None):
     """Convert a technology file to edb control file (xml).
 
@@ -1255,7 +1253,6 @@ class ControlFile:
                                                 )
         return True
 
-    @pyedb_function_handler()
     def write_xml(self, xml_output):
         """Write xml to output file
 

pyedb/dotnet/edb_core/edb_data/hfss_extent_info.py

@@ -23,7 +23,6 @@
 from pyedb.dotnet.edb_core.edb_data.edbvalue import EdbValue
 from pyedb.dotnet.edb_core.edb_data.primitives_data import cast
 from pyedb.dotnet.edb_core.general import convert_pytuple_to_nettuple
-from pyedb.generic.general_methods import pyedb_function_handler
 
 
 class HfssExtentInfo:
@@ -49,12 +48,10 @@ class HfssExtentInfo:
             "pml": self._pedb.edb_api.utility.utility.OpenRegionType.PML,
         }
 
-    @pyedb_function_handler()
     def _get_edb_value(self, value):
         """Get EDB value."""
         return self._pedb.edb_value(value)
 
-    @pyedb_function_handler()
     def _update_hfss_extent_info(self, hfss_extent_info):
         return self._pedb.active_cell.SetHFSSExtentInfo(hfss_extent_info)
 
@@ -321,7 +318,6 @@ class HfssExtentInfo:
         info.UseXYDataExtentForVerticalExpansion = value
         self._update_hfss_extent_info(info)
 
-    @pyedb_function_handler()
     def load_config(self, config):
         """Load HFSS extent configuration.
 
@@ -334,7 +330,6 @@ class HfssExtentInfo:
             if hasattr(self, i):
                 setattr(self, i, j)
 
-    @pyedb_function_handler()
     def export_config(self):
         """Export HFSS extent information.
 

pyedb/dotnet/edb_core/edb_data/hfss_pi_simulation_setup_data.py

@@ -21,15 +21,12 @@
 # SOFTWARE.
 
 from pyedb.dotnet.edb_core.general import convert_py_list_to_net_list
-from pyedb.dotnet.edb_core.utilities.simulation_setup import (
-    BaseSimulationSetup,
-    EdbFrequencySweep,
-)
-from pyedb.generic.data_handlers import pyedb_function_handler
+from pyedb.dotnet.edb_core.sim_setup_data.data.sweep_data import SweepData
+from pyedb.dotnet.edb_core.utilities.simulation_setup import SimulationSetup
 from pyedb.generic.general_methods import generate_unique_name
 
 
-class HFSSPISimulationSetup(BaseSimulationSetup):
+class HFSSPISimulationSetup(SimulationSetup):
     """Manages EDB methods for HFSSPI simulation setup."""
 
     def __init__(self, pedb, edb_object=None):
@@ -39,7 +36,6 @@ class HFSSPISimulationSetup(BaseSimulationSetup):
         self._edb_setup_info = None
         self.logger = self._pedb.logger
 
-    @pyedb_function_handler
     def create(self, name=None):
         """Create an HFSS setup."""
         self._name = name
@@ -76,7 +72,6 @@ class HFSSPISimulationSetup(BaseSimulationSetup):
         else:
             self.logger.error(f"Property position expects an integer value while the provided value is {value}.")
 
-    @pyedb_function_handler()
     def add_frequency_sweep(self, name=None, frequency_sweep=None):
         """Add frequency sweep.
 
@@ -106,7 +101,7 @@ class HFSSPISimulationSetup(BaseSimulationSetup):
             return False
         if not name:
             name = generate_unique_name("sweep")
-        return EdbFrequencySweep(self, frequency_sweep, name)
+        return SweepData(self, frequency_sweep, name)
 
 
 class HFSSPISimulationSettings(object):