From 11df64e463c440b40225cd5da134f42856448158 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Sch=C3=B6nfeldt?= <patrik.schoenfeldt@dlr.de>
Date: Wed, 11 Oct 2023 10:16:46 +0200
Subject: [PATCH] Expose Entity in API

It might make sense to have the abstract superclass of
Node and Edge exposed in the API.
---
 src/oemof/network/__init__.py         | 2 ++
 src/oemof/network/network/__init__.py | 2 ++
 2 files changed, 4 insertions(+)

diff --git a/src/oemof/network/__init__.py b/src/oemof/network/__init__.py
index d606b56..73b1571 100644
--- a/src/oemof/network/__init__.py
+++ b/src/oemof/network/__init__.py
@@ -7,6 +7,7 @@
 from .network import Bus
 from .network import Component
 from .network import Edge
+from .network import Entity
 from .network import Node
 from .network import Sink
 from .network import Source
@@ -16,6 +17,7 @@
     "Bus",
     "Component",
     "Edge",
+    "Entity",
     "energy_system",
     "graph",
     "groupings",
diff --git a/src/oemof/network/network/__init__.py b/src/oemof/network/network/__init__.py
index 137ad65..9d5fc72 100644
--- a/src/oemof/network/network/__init__.py
+++ b/src/oemof/network/network/__init__.py
@@ -1,4 +1,5 @@
 from .edge import Edge
+from .entity import Entity
 from .nodes import Bus
 from .nodes import Component
 from .nodes import Node
@@ -10,6 +11,7 @@
     "Bus",
     "Component",
     "Edge",
+    "Entity",
     "Node",
     "Sink",
     "Source",