Merge branch 'fix-file-version-from-string'

tmontaigu · Jan 9, 2021 · 3f0d6e7 · 3f0d6e7
2 parents 7cf17de + 2512e18
commit 3f0d6e7
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/pylas/header.py b/pylas/header.py
@@ -122,6 +122,11 @@ def __init__(
         version: Optional[Version] = None,
         point_format: Optional[PointFormat] = None,
     ) -> None:
+        if isinstance(point_format, int):
+            point_format = PointFormat(point_format)
+        if isinstance(version, str):
+            version = Version.from_str(version)
+
         if version is None and point_format is None:
             version = LasHeader.DEFAULT_VERSION
             point_format = LasHeader.DEFAULT_POINT_FORMAT

diff --git a/pylas/lib.py b/pylas/lib.py
@@ -223,9 +223,6 @@ def create_las(
        A new las data object
 
     """
-    if isinstance(point_format, int):
-        point_format = PointFormat(point_format)
-
     header = LasHeader(point_format=point_format, version=file_version)
     return LasData(header=header)