LocalDataset: Store remote dataset id in the manifest

src/fairly/dataset/local.py

@@ -397,7 +397,8 @@ def upload(self, repository=None, notify: Callable=None, strategy: str="auto", f
             raise ValueError("Invalid repository")
 
         # Prevent upload if a remote version exists and upload is not enforced
-        if client.id in self.remote_datasets and not force:
+        if client.repository_id in self.remote_datasets and not force:
+            # TODO: Check if remote dataset is valid, otherwise force upload
             raise Warning("Remote dataset exists")
 
         # Create dataset
@@ -484,6 +485,14 @@ def upload(self, repository=None, notify: Callable=None, strategy: str="auto", f
             client.delete_dataset(dataset.id)
             raise
 
+        # Add remote dataset id to the manifest if known repository
+        if client.repository_id:
+            manifest = self._get_manifest()
+            if "remotes" not in manifest:
+                manifest["remotes"] = {}
+            manifest["remotes"][client.repository_id] = dataset.id
+            self._set_manifest(manifest)
+
         return dataset