typing

src/ifcopenshell-python/ifcopenshell/api/sequence/add_work_time.py

@@ -15,9 +15,12 @@
 #
 # You should have received a copy of the GNU Lesser General Public License
 # along with IfcOpenShell.  If not, see <http://www.gnu.org/licenses/>.
+import ifcopenshell
 
 
-def add_work_time(file, work_calendar=None, time_type="WorkingTimes") -> None:
+def add_work_time(
+    file: ifcopenshell.file, work_calendar: ifcopenshell.entity_instance, time_type: str = "WorkingTimes"
+) -> ifcopenshell.entity_instance:
     """Add either working times or holiday times to a calendar
 
     A calendar defines when work occurs by defining working times and