From 45d64e5b82445b97b6e2b62c449d4d0824dcbe66 Mon Sep 17 00:00:00 2001
From: Si11ium <steffen.illium@ifi.lmu.de>
Date: Fri, 2 Aug 2019 13:13:43 +0200
Subject: [PATCH] Now with included sorting per cloud variation - Some
 korrektions

---
 dataset/shapenet.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/dataset/shapenet.py b/dataset/shapenet.py
index 6d6cd57..7f0171a 100644
--- a/dataset/shapenet.py
+++ b/dataset/shapenet.py
@@ -24,6 +24,7 @@ class CustomShapeNet(InMemoryDataset):
 
     def __init__(self, root, collate_per_segment=True, train=True, transform=None, pre_filter=None, pre_transform=None,
                  headers=True, has_variations=False):
+        assert not (collate_per_segment and has_variations), 'Either use each element or pointclouds - with variations'
         self.has_headers = headers
         self.has_variations = has_variations
         self.collate_per_element = collate_per_segment