Remove TODO usage entities in message

Author: jpraynaud

mithril-common/src/messages/certificate.rs

@@ -1,10 +1,8 @@
 use serde::{Deserialize, Serialize};
 
-// TODO: We should probably not rely on entities when defining the message
 use crate::entities::{Beacon, CertificateMetadata, ProtocolMessage};
 
 #[cfg(any(test, feature = "test_only"))]
-// TODO: We should probably not rely on entities when defining the message
 use crate::entities::{ProtocolMessagePartKey, ProtocolParameters, SignerWithStake};
 
 /// Message structure of a certificate

mithril-common/src/messages/snapshot.rs

@@ -1,9 +1,7 @@
 use serde::{Deserialize, Serialize};
 
-// TODO: We should probably not rely on entities when defining the message
 use crate::entities::Beacon;
 
-// TODO: We should probably not rely on entities when defining the message
 #[cfg(any(test, feature = "test_only"))]
 use crate::entities::Epoch;
 /// Message structure of a snapshot

mithril-common/src/messages/snapshot_list.rs

@@ -1,11 +1,10 @@
 use serde::{Deserialize, Serialize};
 
-// TODO: We should probably not rely on entities when defining the message
 use crate::entities::Beacon;
 
-// TODO: We should probably not rely on entities when defining the message
 #[cfg(any(test, feature = "test_only"))]
 use crate::entities::Epoch;
+
 /// Message structure of a snapshot list
 pub type SnapshotListMessage = Vec<SnapshotListItemMessage>;